1. 程式人生 > >python leetcode 386. Lexicographical Numbers

python leetcode 386. Lexicographical Numbers

先了解什麼是字典順序 比如201
[1,10,100,101,102,103,104,105,106,107,108,109,11,110,…198,199,2,20,200,201,21,…]

可以用sorted函式 但是時間複雜度會增加
觀察整理規律可得
cur為當前數字
1.若cur10<=n 則下個數為cur10 否則
2.如果cur數字末尾不為9並且cur+1不大於n 則cur+=1 否則
3.觀察不是2的情況 (i)如果cur=n 可以看到201 下一個數為21 cur=cur//10+1
(ii)如果cur數字末尾為9 109->11 199->2 也就是去cur的末尾數字直到不為9 再 cur=cur//10+1

class Solution:
    def lexicalOrder(self, n):
        """
        :type n: int
        :rtype: List[int]
        """
        if n<1:
            return []
        cur=1
        res=[]
        for i in range(n):
            res.append(cur)
            if cur*10<=n:
                cur=cur*10 
            elif cur%10!=9 and cur+1<=n:
                cur+=1 
            else:
                while ((cur//10) % 10)==9:
                    cur=cur//10
                cur=cur//10+1 
        return res