力扣 题目40-- 组合总和 II

题目

---

题解

---

 

在力扣 题目39-- 组合总和中我们做过类似的题

而40题多加了条件 即 有重复数字而且每个数只能选一次

其实我们可以从39题条件转换一下,即无限制重复被选取->有限制重复被选取

例如 candidates = [2,5,2,1,2]

就是 1可以取1次 2可以取3次 5可以取1次

这样的话就简单了 我们只要在39题的基础上加入限制数量 于是稍加改造即可

使用map容器 key做为数字 value作为数量 而且map也可以自动排序 

在39题的基础上加入

如果value为0则不选

如果value不为0则选 并且value-1

代码

#include<iostream>
#include<algorithm>
#include<vector>
#include<map>
using namespace std;
int binationSum(map<int, int>& m, int &target, map<int, int>::iterator position, int add, vector<int>& test, vector<vector<int>>& res)
{
    //迭代器遍历
    for (map<int, int>::iterator it = position; it != m.end(); it++)
    {
        //四种情况 1.如果当前数字的数量为0 则不取 跳过
        if (it->second == 0) {
            continue;
        }
        //四种情况 2.小于 add是上一轮给的值 如果这个数+it->first小于了 说明有可能成为解 更新一下add 继续递归
        else if (add + it->first < target) {
            //test变更
            test.push_back(it->first);
            //数量变更
            it->second -= 1;
            binationSum(m, target, it, add + (it->first), test, res);
            //回溯
            it->second += 1;
            test.pop_back();
        }
        //四种情况 3.等于 add是上一轮给的值 如果这个数+it->first等于了 说明test加上it->first就是解 直接给res
        else if(add + it->first == target)
        {
            test.push_back(it->first);
            res.push_back(test);
            test.pop_back();
        }
        //四种情况 4.大于 add是上一轮给的值 如果这个数 + it->first大于了 说明无解 直接返回 退出
        else {
            return 0;
        }
    }
    return 0;
}

class Solution {
public:
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        //把candidates放入map中
        map<int, int> m;
        for (int i = 0; i < candidates.size(); i++) {
            auto mm = m.find(candidates[i]);
            if (mm == m.end()) {
                m.insert(pair<int, int>(candidates[i], 1));
            }
            else
            {
                mm->second += 1;
            }
        }
        vector<vector<int>>  res;
        vector<int> test;
        //调用
        binationSum(m, target, m.begin(), 0, test, res);
        return res;
    }
};
int main() {
    Solution sol;
    vector<int> candidates = { 2,5,2,1,2 };
    int target = 5;
    vector<vector<int>>  res = sol.combinationSum2(candidates, target);
    for (int i = 0; i < res.size(); i++) {
        for (int j = 0; j < res[i].size(); j++) {
            cout << res[i][j];
        }
        cout << endl;
    }
}