LeetCode筆記:386. Lexicographical Numbers

問題:

Given an integer n, return 1 - n in lexicographical order.
For example, given 13, return: [1,10,11,12,13,2,3,4,5,6,7,8,9].
Please optimize your algorithm to use less time and space. The input size may be as large as 5,000,000.

思路:

給出一個整數n,返回 1 到 n 的詞典順序。
比如,給出13,返回 [1,10,11,12,13,2,3,4,5,6,7,8,9]。
請優化你的代碼來使用更少的時間和空間。輸入范圍會達到5000000。

思路:

問題的關鍵在于理清“詞典順序”是什么樣的。

詞典順序就是10比2靠前,101比11靠前,110比11靠后,你可以簡單的嘗試一下把數字都轉換成字符串數組,然后對字符串數組進行 Arrays.sort(),就會得到字符串下詞典順序的數字。但是這個方法因為用了排序,是會超時的。

所以我們的做法大致順序如下:

1、如果一個數乘以十以后沒有超過n,那它后面緊挨著的應該是它的十倍,比如1,10,100。
2、如果不滿足1,那就應該是直接加一,比如n為13的時候,前一個數為12,120超過了n,那接著的應該是13。但是這里要注意如果前一個數的個位已經是9或者是它就是n了,那就不能加一了,比如 n = 25,前一個數為19,下一個數應該為2而不是19+1=20。25的下一個也沒有26了。
3、如果不滿足2,比如19后面應該接2而不是20,這時候應該將19除以10再加一,比如n=500,399的下一個應該是4,那就是除以十,個位還是9,繼續除以10,得到3,加一得到4。

將上面的過程整理成代碼就可以了,循環的次數就是n,也就是總個數。

代碼(Java):

public class Solution {
    public List<Integer> lexicalOrder(int n) {
        List<Integer> res = new ArrayList<Integer>();
        int cur = 1;
        for (int i = 1; i <= n; i++) {
            res.add(cur);
            if (cur * 10 <= n) {
                cur = cur * 10;
            } else if (cur % 10 != 9 && cur + 1 <= n) {
                cur++;
            } else {
                while ((cur / 10) % 10 == 9) {
                    cur = cur / 10;
                }
                cur = cur / 10 + 1;
            }
        }
        return res;
    }
}

合集:https://github.com/Cloudox/LeetCode-Record


查看作者首頁

?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容