a=[1,2,3] n=len(a) res=[] def permutation(a,solution): #注意,這里要用global修飾res,才能更新結果 global res if len(a)==0: res.append(solution) return for i in range(len(a)):
newsolution=solution+[a[i]] new_a=a[:i]+a[i+1:] permutation(new_a,newsolution) permutation(a,[]) print(res) 輸出: [[1, 2, 3], [1, 3, 2], [2, 1, 3], [2, 3, 1], [3, 1, 2], [3, 2, 1]]
基本思路:
其實對於回溯法,我們要從反向開始考慮。我們每次從原始數組中選擇一個加入到結果中,當原始數組中(新建的)沒有元素時(也就是len(a)==0,此時結果為[1,2,3]),我們得到了第一個排列,我們將這個排列加入到結果集中,然后返回上一步,也就是我們現在有[1,2],再返回一步[1],此時再加入3,再加入2,得到[1,3,2],
以此類推。