LeetCode-90.子集II(相关话题:回溯)

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

说明:解集不能包含重复的子集。

示例:

输入: [1,2,2]
输出:
[
  [2],
  [1],
  [1,2,2],
  [2,2],
  [1,2],
  []
]

解题思路:先将数组nums排序,然后逐个求解元素个数为0-nums.length的所有子集

Java代码:

class Solution {
    public List<List<Integer>> subsetsWithDup(int[] nums) {
        Arrays.sort(nums);
        List<List<Integer>> res = new LinkedList<>();
        for(int i = 0; i <= nums.length; i++){
            List<Integer> tmp = new ArrayList<>(i);
            dfs(nums, 0, i, tmp, res);
        }

        return res;
    }
    private void dfs(int[] nums, int s, int n, List<Integer> tmp, List<List<Integer>> res){
        if(tmp.size() == n){
            res.add(new ArrayList<Integer>(tmp));
            return;
        }

        for(int i = s; i < nums.length; i++){
            if(i > s && nums[i] == nums[i-1])
                continue;

            tmp.add(nums[i]);
            dfs(nums, i+1, n, tmp, res);
            tmp.remove(tmp.size()-1);
        }
    }
}

猜你喜欢

转载自blog.csdn.net/weixin_38823568/article/details/82844974