LeetCode 刷题系列 -- 90. 子集 II

给你一个整数数组 nums ,其中可能包含重复元素,请你返回该数组所有可能的子集(幂集)。

解集 不能 包含重复的子集。返回的解集中,子集可以按 任意顺序 排列。

示例 1:

输入:nums = [1,2,2]
输出:[[],[1],[1,2],[1,2,2],[2],[2,2]]
示例 2:

输入:nums = [0]
输出:[[],[0]]
 

提示:

1 <= nums.length <= 10
-10 <= nums[i] <= 10

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/subsets-ii
 

思路:回溯算法秒杀所有排列/组合/子集问题 :: labuladong的算法小抄

回溯算法。此类型为 元素可重不可复选。
1. 先将数组中的元素排序
2. 定义数组  used ,长度为  nums.length , used[i] 代表 nums[i] 是否被放入子集中
3. 回溯算法遍历数组。注意剪枝问题。

java代码:

class Solution {
 public List> subsetsWithDup(int[] nums) {
        List> result = new LinkedList<>();
        boolean[] used = new boolean[nums.length];
        
        Arrays.sort(nums);
        backtrack(nums,0,used,new LinkedList<>(),result);
        
        return result;
    }

    public void backtrack(int[] nums, int start,boolean[] used, List subSet, List> result) {

        result.add(new LinkedList<>(subSet));
        
        for (int i = start; i < nums.length; i++) {
            
 // 剪枝, 若是当前元素与前一个元素相等,且前一个元素没使用过,那说明需要剪枝,因为此时进入到了新的周期,一个与之前相等元素相同的周期。
            // 具体剪枝可以看二叉树            if(i>0&&nums[i-1] == nums[i] && !used[i-1]) {
                continue;
            }
            
            subSet.add(nums[i]);
            used[i] = true;
            backtrack(nums, i + 1,used, subSet, result);
            subSet.remove(subSet.size() - 1);
            used[i] = false;
        }

    }
}

你可能感兴趣的:(回溯算法,leetcode,算法,java)