LeetCode40-组合总和 II

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

candidates 中的每个数字在每个组合中只能使用一次。

说明:

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

示例 1:

输入: candidates = [10,1,2,7,6,1,5], target = 8,
所求解集为:
[
  [1, 7],
  [1, 2, 5],
  [2, 6],
  [1, 1, 6]
]

示例 2:

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

一、思路

这题和LeetCode39很像,应该用回溯算法也能解决

(一)回溯算法

回溯算法最重要的是确定什么时候回溯。
这里,回溯的条件是:总和大于或等于target

算法流程:
(1)对数组排序
(2)从大的数开始加起,每次相加后与target进行比较

  • 若大于target:取下一个更小的数
  • 若等于target:保存

C++代码:

class Solution {
public:
	set<vector<int>> results;
	int tar;
	vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
		if (candidates.empty())
			return vector<vector<int>>(results.begin(), results.end());

		sort(candidates.begin(), candidates.end());
		vector<int> res;
		tar = target;

		recall(candidates, candidates.size(), 0, res);
		return vector<vector<int>>(results.begin(), results.end());
	}

	void recall(vector<int>& candidates, int pre, int sum, vector<int> res) {
		if (sum == tar) {
			results.insert(res);
			return;
		}

		if (sum > tar)
			return;

		for (int i = pre - 1; i >= 0; i--) {
			sum += candidates[i];
			res.push_back(candidates[i]);
			recall(candidates, i, sum, res);
			res.pop_back();
			sum -= candidates[i];
		}
	}
};

执行效率:
LeetCode40-组合总和 II

相关文章: