LeetCode - 子集II

题目链接:https://leetcode-cn.com/problems/subsets-ii/

题目描述

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

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

示例:

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

我的思路

  • 是LeetCode-子集的加强版,增加了一些判断条件,整体的思路没变。
  • 同样使用回溯法,属于没有递归基的递归,每调用一次递归,就输出当前 temp 中的元素。例如本题,元素弹出的顺序为 1 | 1,2 | 1,2,2 | 1,2 | 2,2 | 2 。递归树如下:
    root
  /	  |	  \
 1    2    2
/ \   |
2 2   2
|
3
  • 但我们会发现,1,2 数组有两个,不符题意。因此每递归到一层,应该去掉该层相同的数字,比如例子中该去掉第二层的第二个2。
  • 还有一点不同,需要在运算前,将原数组进行排序。

代码如下:

class Solution {
    vector<vector<int>> res;
    vector<int> temp;
public:
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        sort(nums.begin(), nums.end());
        recur(nums, 0);
        return res;
    }
    void recur(vector<int>& nums, int index){
        res.push_back(temp);
        for(int i = index; i < nums.size(); i++){
            if(i > index && nums[i] == nums[i-1]) continue;	//去掉该层相同的数字
            temp.push_back(nums[i]);
            recur(nums, i + 1);
            temp.pop_back();
        }
        return;
    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章