LeetCode 40. Combination Sum II

来源:互联网 发布:双录取很坑人 知乎 编辑:程序博客网 时间:2024/06/05 07:23

题意

给定一个可能有重复元素的序列,找出取出的数之和为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();            }        }    }};
原创粉丝点击