LeetCode 40. Combination Sum II

题意

给定一个可能有重复元素的序列，找出取出的数之和为$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();
            }
        }
    }
};