Problem 1: Permutation I
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].
分析:
这次,我们要生成完整的序列了,那么和上次的Subset有什么不同呢?
1. 添加进res的时间,上面题我们每添加一个数字到tmp里,就可以往res里面添加,而这次,我们要生成完整的序列,所以需要当tmp.size()==序列长度的时候再往res里面添加
2. 每次不能从pos开始往数组尾巴扫了,因为我们要求的不是Subset而是完整序列,所以要从第一个数字开始往数组尾巴扫,问题又来了,我们怎么知道取没取过这个元素呢,那么我们就创建一个boolean[] visit 每此添加的时候给相对应位置置True,删去的时候置False
public class Solution { public ArrayList<ArrayList<Integer>> permute(int[] num) { ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<Integer>>(); ArrayList<Integer> tmp = new ArrayList<Integer>(); boolean[] visit = new boolean[num.length]; Arrays.sort(num); dfs(res,tmp,num,visit); return res; } public void dfs(ArrayList<ArrayList<Integer>> res, ArrayList<Integer> tmp, int[] num, boolean[] visit){ if(tmp.size()==num.length) { res.add(new ArrayList<Integer>(tmp)); return; } for(int i=0; i<num.length; i++){ if(visit[i]==false){ tmp.add(num[i]); visit[i] = true; dfs(res,tmp,num,visit); tmp.remove(tmp.size()-1); visit[i] = false; } } } }
几点注意的地方:
1. 每次判断是否visit的时候,这个if语句应该概括for循环里所有的动作,因为这么想,如果visit过,我们什么事也不做,直接跳过
2. 给res添加完之后,别忘了return
Problem 2 Permutation II
Given a collection of numbers that might contain duplicates, return all possible unique permutations. For example, [1,1,2] have the following unique permutations: [1,1,2], [1,2,1], and [2,1,1].
同样的思路,有重复的数字出现在集合里,而不能生成重复的序列
我们运用和Subset II 一样的思路,在remove后直接找到下一个和这次添加不重复的数字
public class Solution { public ArrayList<ArrayList<Integer>> permuteUnique(int[] num) { ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<Integer>>(); ArrayList<Integer> tmp = new ArrayList<Integer>(); boolean[] visit = new boolean[num.length]; Arrays.sort(num); dfs(res,tmp,num,visit); return res; } public void dfs(ArrayList<ArrayList<Integer>> res, ArrayList<Integer> tmp, int[] num, boolean[] visit){ if(tmp.size()==num.length) { res.add(new ArrayList<Integer>(tmp)); return; } for(int i=0; i<num.length; i++){ if(visit[i]==false){ tmp.add(num[i]); visit[i] = true; dfs(res,tmp,num,visit); tmp.remove(tmp.size()-1); visit[i] = false; while(i<num.length-1 && num[i] == num[i+1]) i++; } } } }