随笔- 260 文章- 3 评论- 31 阅读- 17万

【leetcode刷提笔记】Permutations

Given a collection of numbers, return all possible permutations.

For example,
[1,2,3] have the following permutations:
[1,2,3], [1,3,2], [2,1,3], [2,3,1], [3,1,2], and [3,2,1].

用深度搜索的方法搜索下面的一棵树（示例，非完整的树），每次搜索到叶子时经过的路径就是一个排列。

代码如下：

 1 public class Solution {
 2     public List<List<Integer>> permute(int[] num) {
 3         List<List<Integer>> answerList = new ArrayList<List<Integer>>();
 4         ArrayList<Integer> currentList = new ArrayList<Integer>();
 5         boolean[] visited = new boolean[num.length];
 6         DS(answerList, visited, num, currentList);        
 7         return answerList;
 8     }
 9     
10     public void DS(List<List<Integer>> answerList,boolean[] visited,int[] num,ArrayList<Integer> currentList){
11         boolean find = true;
12         for(int i = 0;i < num.length;i++){
13             if(!visited[i]){
14                 currentList.add(num[i]);
15                 visited[i]= true;
16                 DS(answerList, visited, num, currentList);
17                 visited[i]= false;
18                 currentList.remove(currentList.size()-1);
19                 find = false;
20             }
21         }
22         if(find){
23             ArrayList <Integer> temp = new ArrayList<Integer>(currentList);
24             answerList.add(temp);
25         }
26     }
27 }

上述代码中DS函数为递归深度优先搜索函数，answerList记录最终得到的所有排列，visited数据记录在某个时间点对应节点是否在访问过的路径上，currentList为当前走过的路径。要注意的一点是找到一条新的路径后，要为这条路径申请内存空间存放它（代码第23行所示），否则currentList被修改的时候已经存放到answerList中的排列也会被修改。