40. Combination Sum II

来源:互联网 发布:广州开发区西区程序员 编辑:程序博客网 时间:2024/06/05 16:33

此题是Combination Sum的扩展,但是每个元素最多只能选用一次。由于有重复元素的存在,比如数组为[1(1),1(2),2,3],target = 6,可能出现重复结果1(1),2,3 和 1(2),2,3。我们可以如下处理:在for循环中,也就是某layer尝试放入下一个数y的时候,我们提前判断下一个数y是否跟当前的数x相等,如果相等,我们就直接跳过去,因为第一次放x的时候,实际上已经包含了conbination包含x的所有可能性,再尝试y就会产生重复。代码如下:

class Solution {private:    vector<vector<int>> res;    void helper(vector<int>& candidates, int target, int index, vector<int>& tmpres) {        if (target == 0) {            res.push_back(tmpres);            return;        }        for (int i=index; i<candidates.size() && target>=candidates[i]; i++) {            tmpres.push_back(candidates[i]);            helper(candidates, target-candidates[i], i+1, tmpres);            tmpres.pop_back();            while (i+1<candidates.size() && candidates[i]==candidates[i+1]) { // 如果此层,下一个数跟当前数相等,则直接跳过,直至i位置的数,是最后一个跟当前数相等的数                i++;            }        }    }public:    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {        vector<int> tmpres;        sort(candidates.begin(), candidates.end());        helper(candidates, target, 0, tmpres);        return res;    }};



0 0
原创粉丝点击