leetcode_subsets-ii

题目链接

https://leetcode-cn.com/problems/subsets-ii/

题目内容

给定一个可能包含重复元素的整数数组 nums，返回该数组所有可能的子集（幂集）。
说明：解集不能包含重复的子集。

示例

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

解题思路

本题采用了深度优先搜索算法。
具体请看代码（有注释）。

代码

class Solution {
public:
    vector<vector<int>> res;
    vector<int> now;
    int n ;
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        n = nums.size();
        dfs(nums,0);
        return res;
    }

    void dfs(vector<int>& nums, int u)
    {
        if (u == n)
        {
            res.push_back(now);
            return ;
        }
        // 
        int k = 0;
        while(u+k < n && nums[u] == nums[u+k])
            k++;
        //对于nums[u]而言，有k个重复的元素，k的取值最小为1
        for (int i = 0; i <= k; i++)
        {
            dfs(nums, u+k);
            now.push_back(nums[u]);
        }
        //恢复现场，now在恢复完现场后被自动清空
        for(int i = 0; i <= k; i++)
            now.pop_back();
    }
};