Java LeetCode 90. 子集 II

给定一个可能包含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
说明:解集不能包含重复的子集。
示例:
输入: [1,2,2]
输出:
[
[2],
[1],
[1,2,2],
[2,2],
[1,2],
[]
]

这道题最重要的是剪枝的思想,首先如果当前遍历元素已经为true,则不用继续往下找了,,如果当前遍历元素和上一个元素相同,且上一个元素没有被使用过,那表示在下面上一个元素一定会出现,这样也不符合要求,也需要剔除

class Solution {
    public List<List<Integer>> subsetsWithDup(int[] nums) {
        Arrays.sort(nums);//首先必须排序
        List<List<Integer>> res = new ArrayList();
        boolean[] pau = new boolean[nums.length];//记录已经遍历过得元素
        Deque<Integer> que = new LinkedList();//存储已经遍历过得元素
        back(nums,res,que,pau,0);
        return res;
    }

    public void back(int[] nums,List<List<Integer>> res,Deque<Integer> que,boolean[] pau,int start){
        
        res.add(new ArrayList(que));
        for(int i=start;i<nums.length;i++){
            if(pau[i]){
                continue;
            }
            if(i>0&&nums[i]==nums[i-1]&&!pau[i-1]){
                continue;
            }

            que.addLast(nums[i]);
            pau[i]=true;

            back(nums,res,que,pau,i+1);

            que.removeLast();
            pau[i]=false;

        }
    }
}

你可能感兴趣的:(LeetCode)