LeetCode 力扣 60. 第k個排列

題目描述(中等難度)

又是一道全排列的題,之前在31題46題,也討論過全排列問題的一些解法。這道題的話,是給一個 n,不是輸出它的全排列,而是把所有組合從從小到大排列后,輸出第 k 個。

解法一

以 n = 4 為例,可以結合下圖看一下。因為是從小到大排列,那么最高位一定是從 1 到 4。然后可以看成一組一組的,我們只需要求出組數,就知道最高位是多少了。而每組的個數就是 n - 1 的階乘,也就是 3 的階乘 6。

算組數的時候, 1 到 5 除以 6 是 0,6 除以 6 是 1,而 6 是屬于第 0 組的,所有要把 k 減去 1。這樣做除法結果就都是 0 了。

int perGroupNum = factorial(n - 1); 
int groupNum = (k - 1) / perGroupNum;

當然,還有一個問題下次 k 是多少了。求組數用的除法,余數就是下次的 k 了。因為 k 是從 1 計數的,所以如果 k 剛好等于了 perGroupNum 的倍數,此時得到的余數是 0 ,而其實由于我們求 groupNum 的時候減 1 了,所以此時 k 應該更新為 perGroupNum。

k = k % perGroupNum; 
k = k == 0 ? perGroupNum : k;

舉個例子,如果 k = 6,那么 groupNum = ( k - 1 ) / 6 = 0, k % perGroupNum = 6 % 6 = 0,而下次的 k ,可以結合上圖,很明顯是 perGroupNum ,依舊是 6。

結合下圖,確定了最高位屬于第 0 組,下邊就和上邊的情況一樣了。唯一不同的地方是最高位是 2 3 4,沒有了 1。所有得到 groupNum 怎么得到最高位需要考慮下。

我們可以用一個 list 從小到大保存 1 到 n,每次選到一個就去掉,這樣就可以得到 groupNum 對應的數字了。

List<Integer> nums = new ArrayList<Integer>();
for (int i = 0; i < n; i++) {
    nums.add(i + 1);
}
int perGroupNum = factorial(n - 1);
int groupNum = (k - 1) / perGroupNum;
int num = nums.get(groupNum); //根據 groupNum 得到當前位
nums.remove(groupNum);//去掉當前數字

綜上,我們把它們整合在一起。

public String getPermutation(int n, int k) {
    List<Integer> nums = new ArrayList<Integer>();
    for (int i = 0; i < n; i++) {
        nums.add(i + 1);
    }
    return getAns(nums, n, k);
}

private String getAns(List<Integer> nums, int n, int k) {
    if (n == 1) {
        //把剩下的最后一個數字返回就可以了
        return nums.get(0) + "";
    }
    int perGroupNum = factorial(n - 1); //每組的個數
    int groupNum = (k - 1) / perGroupNum;
    int num = nums.get(groupNum);
    nums.remove(groupNum);
    k = k % perGroupNum; //更新下次的 k 
    k = k == 0 ? perGroupNum : k;
    return num + getAns(nums, n - 1, k);
}
public int factorial(int number) {
    if (number <= 1)
        return 1;
    else
        return number * factorial(number - 1);
}

時間復雜度:

空間復雜度:

這是最開始自己的想法,有 3 點可以改進一下。

第 1 點,更新 k 的時候,有一句

k = k % perGroupNum; //更新下次的 k 
k = k == 0 ? perGroupNum : k;

很不優雅了,問題的根源就在于問題給定的 k 是從 1 編碼的。我們只要把 k - 1 % perGroupNum,這樣得到的結果就是 k 從 0 編碼的了。然后求 groupNum = (k - 1) / perGroupNum; 這里 k 也不用減 1 了。

第 2 點,這個算法很容易改成改成迭代的寫法,只需要把遞歸的函數參數, 在每次迭代更新就夠了。

第 3 點,我們求 perGroupNum 的時候,每次都調用了求迭代的函數,其實沒有必要的,我們只需要一次循環求出 n 的階乘。然后在每次迭代中除以 nums 的剩余個數就夠了。

綜上,看一下優化過的代碼吧。

public String getPermutation(int n, int k) {
    List<Integer> nums = new ArrayList<Integer>();
    int factorial = 1;
    for (int i = 0; i < n; i++) {
        nums.add(i + 1);
        if (i != 0) {
            factorial *= i;
        }
    }
    factorial *= n; //先求出 n 的階乘
    StringBuilder ans = new StringBuilder();
    k = k - 1; // k 變為 k - 1
    for (int i = n; i > 0; i--) { 
        factorial /= (nums.size()); //更新為 n - 1 的階乘
        int groupNum = k / factorial;
        int num = nums.get(groupNum);
        nums.remove(groupNum);
        k = k % factorial;
        ans.append(num);  

    }
    return ans.toString();
}

時間復雜度:O(n),當然如果 remove 函數的時間是復雜度是 O(n),那么整體上就是 O(n2)。

空間復雜度:O(1)。

這道題其實如果寫出來,也不算難,優化的思路可以了解一下。

更多詳細通俗題解詳見 leetcode.wang

?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 228,606評論 6 533
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 98,582評論 3 418
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 176,540評論 0 376
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 63,028評論 1 314
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,801評論 6 410
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 55,223評論 1 324
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,294評論 3 442
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,442評論 0 289
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 48,976評論 1 335
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 40,800評論 3 354
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 42,996評論 1 369
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,543評論 5 360
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,233評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,662評論 0 26
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,926評論 1 286
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,702評論 3 392
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 47,991評論 2 374

推薦閱讀更多精彩內容

  • 在C語言中,五種基本數據類型存儲空間長度的排列順序是: A)char B)char=int<=float C)ch...
    夏天再來閱讀 3,371評論 0 2
  • "use strict";function _classCallCheck(e,t){if(!(e instanc...
    久些閱讀 2,042評論 0 2
  • 算法和數據結構 [TOC] 算法 函數的增長 漸近記號 用來描述算法漸近運行時間的記號,根據定義域為自然數集$N=...
    wxainn閱讀 1,076評論 0 0
  • 計算機二級C語言上機題庫(南開版) 1.m個人的成績存放在score數組中,請編寫函數fun,它的功能是:將低于平...
    MrSunbeam閱讀 6,433評論 1 42
  • 小時候,在家人的懷抱里天真無邪的活著;長大后,發現自己要走出這片天地,勇敢的活著。然而越長大是越孤單的,因為你不知...
    兔子愛上橘子閱讀 179評論 0 0