leetcode–demo65全排列2
这题与上一条的区别就在于列表中可能会出现重复元素。所以我们在处理这一题的时候要学会剪枝。比如列表是[1,1,2]的情况,如果第一次选择是以1开头,那么第二次选择时就需要把以1开的情况给砍掉。因为这两种情况是重复的。代码就是在64题的基础上添加一行if语句,目的是在回溯完成后进行下一次分支选择时,将重复分支砍掉。当然也可以先得到所有结果然后再删除重复的列表。
class Solution {
public List<List<Integer>> permuteUnique(int[] nums) {
//结果列表用于返回结果
List<List<Integer>> res=new ArrayList<>();
//路径列表用于向结果列表添加子列表
List<Integer> path=new ArrayList<>();
if (nums.length==0){
return res;
}
Arrays.sort(nums);
//状态数组用于判断该数是否已经使用过
boolean[] used=new boolean[nums.length];
dfs(res,path,used,nums,0);
return res;
}
private void dfs(List<List<Integer>> res, List<Integer> path, boolean[] used, int[] nums, int depth) {
if (depth==nums.length){
res.add(new ArrayList<>(path));
return;
}
for (int i=0;i<nums.length;i++){
if (used[i]){
continue;
}
if (i>0&&nums[i]==nums[i-1]&&!used[i-1]){
continue;
}
path.add(nums[i]);
used[i]=true;
dfs(res,path,used,nums,depth+1);
//回溯部分
used[i]=false;
path.remove(path.size()-1);
}
}
}