去重列表全排列問題- python解法

昨天 下午朋友發了我一道LeetCode面試題:

給定一個沒有重復的數字序列,返回其所有可能的全排列。示例:
輸入: [1,2,3]
輸出:
[
[1,2,3],
[1,3,2],
[2,1,3],
[2,3,1],
[3,1,2],
[3,2,1]
]

我一看,這很簡單啊,而且題目要求也很清晰,就打算五分鐘之內寫完發給他。
然而寫了二十分鐘,居然,沒有,寫,出,來!

我好菜啊…… (果然算法題是我的噩夢)

網上看了一圈,發現別人的解法涉及到交換位置,講的也不太清楚的樣子。下班在地鐵上,我就開動了并不聰明的腦筋,好好想了一下。

明確問題

全排列是一道高中數學必備的排列組合問題。我們知道:對一個去重列表的所有的可能情況,是 n!
假設列表有六個元素,那么第一個元素的選擇可能有六種情況,第二個有剩下的五種,第三個有四種……然后依次相乘得到所有的情況數。

或者從相反方向來建立,如果列表中只有一個數,那么只有唯一一種排列方式;
如果有兩個數字ab,那么就有a,b 和b,a兩種排列方式;
這個時候,可以認為是結果的第一位有兩種選擇:選a或者b,剩下的一位就是上面的一種排列方式。
這么看可能還不夠直觀,接下來我們看abc三個數字的情況:
第一位有a,b,c三種可能性,我們如果挑出了a放在結果第一位上,剩下的后兩位就是b,c的組合結果,由上述兩種數字排列可知,此時二,三位可以是b,c或者c,b排列,這就是兩種情況;接下來把第一位換成b,剩下兩位就是a,c或者c,a排列;選擇c同理……
如果有abcd四個數字,那么第一位可以選擇abcd任意一個,剩下三位就是剩下三個數字在上一步的全排列。
實際上,我們全排列,就是在n-1的那一層排列情況上再構建了一層,第一位的數字永遠有n種可能性,剩下的幾位就是上一步推出的n-1層排列組合結果,填進去就好了。

需要用到遞歸解法。遞歸的終點是:在只剩一個數字的情況下返回這個數字,列表為空返回空。

代碼如下:

import copy

def permuation(x: list):
    result = []
    if not x:
        return
    if len(x) == 1:
        return [x]

    for num in x:
        new_list = copy.copy(x)
        new_list.remove(num)
        for l in permuation(new_list):
            if l:
                res = [num]
                res.extend(l)
                result.append(res)
    return result

print(permutation([1,2,3]))
>> [[1, 2, 3], [1, 3, 2], [2, 1, 3], [2, 3, 1], [3, 1, 2], [3, 2, 1]]

python由于自身特性,對于列表list的操作時一定要牢記:list在python中是可變對象,意味著如果想要給新列表賦值,需要用到copy來實現復制已有列表的功能,否則用簡單的a = b就會把同一個列表指向a,這是我們不想看到的。
此外,列表的添加項append,擴展extend,刪除項remove 等操作,都是在原列表上進行的,所以不能用賦值,否則會返回空None。而且,代碼里的這一段:

      res = [num]
      res.extend(l)
      result.append(res)

如果想要(偷懶)簡化成:

      result.append([num].extend(l))

也會得到None,所以需要先創建結果子列表再添加。

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