LeetCode47-全排列II-剪枝逻辑

阿维的博客日记2023-11-19 21:54

这里第一层,used只有一个元素为1,代表只取出了1个元素作为排列,第二层used有两个元素为1,代表取出了2个元素作为排列,因为数组有序,所以重复的元素都是挨着的,因此可以使用如下语句去重.

其中visit[i-1]==False的话,就是代表了树层visit[i-1]使用过

其中visit[i-1]==True的话,就是代表了树枝visit[i-1]使用过

java 复制代码

if(i>=1&&nums[i-1]==nums[i]&&!visit[i-1]){
    continue;
}

因为去重的逻辑是减去树层的重复项,因此当visit[i-1]==False的时候必须要跳过,也就是!visit[i-1]的时候要continue,不能是break,如果break了,下一个排列会被忽略掉了!

java 复制代码

class Solution {
    public List<List<Integer>> permuteUnique(int[] nums) {
        List<List<Integer>> paths=new ArrayList<>();
        Deque<Integer> path=new ArrayDeque<>();
        Arrays.sort(nums);
        boolean[] visit = new boolean[nums.length+1];
        dfs(nums,paths,path,visit);
        return paths;
    }

    public void dfs(int[] nums,List<List<Integer>> paths,Deque<Integer> path,boolean[] visit){
        if(nums==null){
            return ;
        }
        if(path.size()==nums.length){
            paths.add(new ArrayList(path));
            return;
        }
        // int i=begin;
        int i=0;
        for(;i<nums.length;++i){
            if(i>=1&&nums[i-1]==nums[i]&&!visit[i-1]){
                continue;
            }
            if(!visit[i]){
                visit[i]=true;
                path.add(nums[i]);
                dfs(nums,paths,path,visit);
                if(!path.isEmpty())
                    path.removeLast();
                visit[i]=false;
            }
        }
    }
}