【回溯】【leetcode】和为target的数的组合(数可重复使用)

题目:

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

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

说明:

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

提示:

  • 1 <= candidates.length <= 30
  • 1 <= candidates[i] <= 200
  • candidate 中的每个元素都是独一无二的。
  • 1 <= target <= 500

例:

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

来源:

39. 组合总和

解题思路:回溯

继续学习回溯的思想,与前面2篇组合不同的是,本题1)数组没有排序,所以先排序;2)数字可重复使用,所以递归时参数不再+1。

  • 递归终止条件:当前sum大于等于target
  • 结果满足条件:sum == target
  • 剪枝条件:当前sum即将超过目标值

c++代码:

class Solution {
public:
    vector< vector<int> > result;
    vector<int> path;
    vector< vector<int> > combinationSum(vector<int>& candidates, int target) {
        sort(candidates.begin(), candidates.end()); // 排序
        back(candidates, target, 0, 0);
        return result;
    }
    void back(const vector<int>& candidates, int target, int sum, int start) {
        if (sum == target) {
            result.push_back(path);
            return;
        }

        for (int i = start; i < candidates.size(); i++) {
            if (sum + candidates[i] > target) break; // 剪枝
            path.push_back(candidates[i]);
            sum += candidates[i];
            back(candidates, target, sum, i); // 因为可以重复使用数字,所以i而不是i+1
            sum -= candidates[i];
            path.pop_back();
        }
    }
};

猜你喜欢

转载自blog.csdn.net/hbuxiaoshe/article/details/114836617