LeetCode 40. Combination Sum II

最新推荐文章于 2020-01-16 23:36:27 发布

AnICoo1

最新推荐文章于 2020-01-16 23:36:27 发布

阅读量383

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签： leetcode 去重 unique

本文链接：https://blog.youkuaiyun.com/helloiamclh/article/details/72848288

LeetCode 专栏收录该内容

74 篇文章

订阅专栏

本文介绍了一种算法，用于从可能含有重复元素的序列中找出所有加和为目标值的组合，并考虑了元素出现次数的限制。通过排序去重及递归搜索实现。

题意

给定一个可能有重复元素的序列，找出取出的数之和为 $target$ 的方案

思路

和上一题差不多，这次里面元素的个数受到了限制，所以可以统计每个数出现的次数，然后对序列进行排序去重，然后用和上一题一样的方法来求，只是中间取出元素的个数不能超过它的个数.

代码

class Solution {
public:
    map<int, int>mp;
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        sort(candidates.begin(), candidates.end());
        size_t len = candidates.size();
        for(int i = 0; i < len; i++){
            mp[candidates[i]]++;
        }
        int n = unique(candidates.begin(), candidates.end()) - candidates.begin();
        vector<int>temp;
        for(int i = 0; i < n; i++){
            temp.push_back(candidates[i]);
        }
        vector<int>now;
        vector<vector<int> >ans;
        findAns(temp, ans, now, 0, target);
        return ans;
    }
private:
    void findAns(vector<int>& temp, vector<vector<int> >& ans, vector<int>& now, int id, int target){
        if(target == 0){
            ans.push_back(now);
            return ;
        }
        if(id >= temp.size()) return ;
        for(int i = 0; i <= mp[temp[id]] && i * temp[id] <= target; i++){
            for(int j = 0; j < i; j++){
                now.push_back(temp[id]);
            }
            findAns(temp, ans, now, id + 1, target - i * temp[id]);
            for(int j = 0; j < i; j++){
                now.pop_back();
            }
        }
    }
};