【leetcode】39 组合总和（回溯）

题目链接：https://leetcode-cn.com/problems/combination-sum/

题目描述

给定一个无重复元素的数组 candidates 和一个目标数 target ，找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的数字可以无限制重复被选取。

说明：

• 所有数字（包括 target）都是正整数。
• 解集不能包含重复的组合。

示例 1:

输入: candidates = [2,3,6,7], target = 7,
所求解集为:
[
  [7],
  [2,2,3]
]

示例 2:

输入: candidates = [2,3,5], target = 8,
所求解集为:
[
  [2,2,2,2],
  [2,3,3],
  [3,5]
]

思路

该问题是一个无限背包的问题，题目要求candidates 中的数字可以无限制重复被选取，并且需要把组合都打印出来，所以我们考虑用回溯方法求解。

（1）我们定义了一个回溯函数 combinationSumCore(vector<int>& candidates, int target, int begin, vector<int>& curr, vector<vector<int>>& ret)，其中begin表示使用字典中元素的起始位置（这样组合的数组元素都是从小到大排列的，避免出现（2,3,3）和（3,2,3）这样重复元素构成的组合）。并且用curr来存储当前的组合。ret存储所有的组合。
（2）当target==0时，表明找到一组解，将curr写入ret；
当target<0，表明当前curr数组和大于target，弹出尾元素，并继续递归；
当target>0，target-candidates[i]作为新target，递归。

代码

class Solution {
public:
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        vector<vector<int>> ret;
        if(candidates.empty()) return ret;
        vector<int> curr;
        combinationSumCore(candidates, target, 0, curr, ret);
        return ret;
    }

    void combinationSumCore(vector<int>& candidates, int target, int begin, vector<int>& curr, vector<vector<int>>& ret){
        if(target == 0){
            ret.push_back(curr);    // curr的和正好等于target
            return;
        }
        else if(target<0)           // curr和大于target
            return;
        for (int i = begin; i < candidates.size(); ++i) {   // 注意用变量begin控制curr数组内的数字都是从小到大
            curr.push_back(candidates[i]);
            combinationSumCore(candidates, target-candidates[i], i, curr, ret); // target-candidates[i]表示选择i元素
            curr.pop_back();
        }
    }
};