Leetcode 39组合总和 C++ 回溯法

题目描述

给定一个无重复元素的数组 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]
]

思路

使用回溯法，递归求解
这里不用事先对其排序。
因为没有重复元素，所以不用担心结果中有重复的组合。

解答

class Solution {
public:
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        vector<vector<int>> res;
        vector<int> mem;
        fun(candidates,target,0,mem,res);
        return res;
    }
    
    void fun(const vector<int> &candidates,int target,int index,vector<int> &mem,vector<vector<int>> &res)
    //这里将mem定义为引用可以减少占用内存，提高速度。不定义为内存也正确
    {
        if(target<0) return;
        if(target==0)
        {
            res.push_back(mem);
            return;
        }
        while(index<candidates.size())
        {
            mem.push_back(candidates[index]);
            fun(candidates,target-candidates[index],index,mem,res);
            mem.pop_back();
            ++index;
        }
    }
};