leetcode--demo65全排列2

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);


        }
    }

}

猜你喜欢

转载自blog.csdn.net/m0_48126147/article/details/112960103