天天看點

leetcode90——Subsets II

題目大意:給出一個含有重複數字的數組,求出它的全部子集,要求不能有重複子集出現

分析:回溯法。和leetcode78類似,但是這道題的數組中包含重複數字,要避免結果集中添加重複答案就需要在目前層避免選擇重複數字。

代碼:

class Solution {
public:
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        vector<vector<int>> ans;
        vector<int> tmp;
        for(int i = 0;i <= nums.size();i++){
            dfs(nums,0,i,tmp,ans);
        }
        return ans;
    }
    void dfs(vector<int>& nums,int cur,int len,vector<int>& tmp,vector<vector<int>>& ans){
        if(tmp.size() == len){
            ans.push_back(tmp);
            return;
        }
        for(int i = cur;i < nums.size();i++){
            tmp.push_back(nums[i]);
            dfs(nums,i + 1,len,tmp,ans);
            tmp.pop_back();
            while(i < nums.size() - 1 && nums[i] == nums[i + 1]) i++;
        }
    }
};