LeetCode40.组合总和2

给定一个数组 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]
]

思路:在90题的基础上进行剪枝,用一个sum来记录当前集合元素的和,当sum > target时,我们直接返回,这就是剪枝,当sum == target且当前item未在result中出现,就将其加入result

#include<iostream>
#include<vector>
#include<set>
#include<algorithm>
using namespace std;

class  Solution{
public:
	vector<vector<int> > combinationSum2(vector<int> &nums,int target) {
		vector<vector<int> > result;
		set<vector<int> > res_set; 
		vector<int> item;
		sort(nums.begin(),nums.end());
		generate(0,target,nums,item,res_set,result);
		return result;			
	} 
	
	void generate(int i,int target,vector<int> &nums,vector<int> &item,set<vector<int>> &res_set,
	vector<vector<int> > &result) {
		if(i == nums.size() || sum > target) {
			return;
		}
		item.push_back(nums[i]);
		sum+=nums[i];
		if(sum == target && res_set.find(item) == res_set.end()) {
			res_set.insert(item);
			result.push_back(item);
		}
		generate(i+1,target,nums,item,res_set,result);	//当前元素加入item,继续向后面所有元素递归		
		item.pop_back();
		sum -= nums[i];	
		generate(i+1,target,nums,item,res_set,result);	//当前元素不加入item,继续向后面所有元素递归
	}
};

int main() {
	vector<int> nums;
	nums.push_back(10);
	nums.push_back(1);
	nums.push_back(2);
	nums.push_back(7);
	nums.push_back(6);
	nums.push_back(1);
	nums.push_back(5);
	vector<vector<int> > result;
	Solution s;
	result = s.combinationSum2(nums,8);
	for(int i = 0; i < result.size(); i++) {
		for(int j = 0; j < result[i].size(); j++) {
			cout<<"["<<result[i][j]<<"]";
		}
		cout<<endl;
	}
	return 0;
}  
©️2020 CSDN 皮肤主题: 数字20 设计师:CSDN官方博客 返回首页