【leetcode】90. Subsets II

最新推荐文章于 2023-12-01 12:28:16 发布

小do在努力

最新推荐文章于 2023-12-01 12:28:16 发布

阅读量545

点赞数

分类专栏：【leetcode】文章标签： leetcode subsets dfs

本文链接：https://blog.csdn.net/zone_programming/article/details/50572900

版权

【leetcode】专栏收录该内容

53 篇文章

订阅专栏

总结一下，leetcode Subsets II ，题目的意思是得到一个含有重复元素的集合的所有子集。

下面分别给出3种解法。

1. dfs. 该解法16ms beats 17.33% of cpp submissions.

class Solution {
public:
    void dfs(int idx, vector<int>& tmp, vector<vector<int>>& ans, vector<int>& nums){
        ans.push_back(tmp);
        
        // for循环的关键是，对于下一个与当前元素相同的元素，都需要跳过
        // 也就是 1, 1, 1, 2, 2, 3, 4, 4. 最终这个for循环只能遍历1, 2, 3, 4.
        for(int i = idx; i < nums.size(); ++ i) {
            if(i && i > idx && nums[i] == nums[i-1]) continue;
            tmp.push_back(nums[i]);
            dfs(i+1, tmp, ans, nums);
            tmp.pop_back();
        }
        
    }
    
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        vector<vector<int>> ans;
        if(nums.size() < 1) return ans;
        // 首先排序
        sort(nums.begin(), nums.end());
        vector<int> tmp;
        dfs(0, tmp, ans, nums);
        return ans;
    }
};

2. map + dfs, 本质上与第一种解法相同，因为map将所有的相同的元素都几种在一个对应关系上，这样就可以直接只遍历一个元素。该解法 8ms beats 75.43% of cpp submissions.

class Solution {
public:
    void dfs(vector<int>& tmp, vector<vector<int>>& ans, map<int, int>& nmap, map<int, int>::iterator begin){
        ans.push_back(tmp);
        
        for(map<int, int>::iterator iter = begin; iter != nmap.end(); ++ iter) {
            if(iter->second){
                tmp.push_back(iter->first);
                (iter->second) --;
                dfs(tmp, ans, nmap, iter);
                (iter->second) ++;
                tmp.pop_back();
            }
            
        }
        
    }
    
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        vector<vector<int>> ans;
        if(nums.size() < 1) return ans;
        map<int, int> nmap;
        for(int i = 0; i < nums.size(); ++ i) {
            nmap[nums[i]] ++;
        }
        vector<int> tmp;
        dfs(tmp, ans, nmap, nmap.begin());
        return ans;
    }
};

3. 利用之前已有的序列，不断的对已有序列进行填充。如 1, 1, 2, 3, 3, 5. ans中初始化的结果是一个空集[]

对于1,1 取出[], 将[1], [1, 1]分别和[]合并，压入结果序列，此时结果序列ans为[], [1], [1,1]

对于2, 取出[], [1], [1,1], 此时将[2]与前3者合并，结果为[2], [1, 2], [1, 1, 2]，加入结果序列ans, 此时结果为[], [1], [1, 1], [2], [1, 2], [1, 1, 2]。

......以此类推。该解法：12ms, beats 22.32% of cpp submissions

class Solution {
public:
    
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        vector<vector<int>> ans;
        if(nums.size() < 1) return ans;
        map<int, int> nmap;
        for(int i = 0; i < nums.size(); ++ i) {
            nmap[nums[i]] ++;
        }
        
        ans.push_back(vector<int>());
        map<int, int>::iterator iter = nmap.begin();
        
        while(iter != nmap.end()) {
            int cnt = iter->second;
            int previousN = ans.size();
            for(int j = 0; j < previousN; ++ j) {
                vector<int> tmp = ans[j];
                for(int k = 0; k < cnt; ++ k) {
                    tmp.push_back(iter->first);
                    ans.push_back(tmp);
                }
            }
            ++ iter;
        }
        
        return ans;
    }
};

与此题类似的题目有 78.subsets 77.combinations