40 组合总和II

莫忘、莫念

已于 2023-02-17 14:21:08 修改

阅读量147

点赞数

分类专栏：回溯文章标签： leetcode 算法职场和发展

于 2023-02-17 13:44:54 首次发布

本文链接：https://blog.csdn.net/m0_63783532/article/details/129085167

版权

回溯专栏收录该内容

3 篇文章 0 订阅

订阅专栏

一、题目

给定一个候选人编号的集合 candidates 和一个目标数 target ，找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的每个数字在每个组合中只能使用一次。

示例 1:

输入: candidates =[10,1,2,7,6,1,5]， target = 8,
输出:

[

[1,1,6]

[1,2,5]

[1,7]

[2,6]

]

二、分析

首先需要将元素排序
然后需要加⼀个bool型数组used，用来记录同⼀树枝上的元素，是否使用过
以num=[1,1,3] ,target=4为例

三、核心代码

重点：

nu[i-1] == nu[i] && u[i-1] == false说明在前一个树枝上使用过nu[i-1]

void BackTrack(int start,int target,int s,vector<int> nu, vector<int>& p,vector<bool>&u, vector<vector<int>>& r)
{
	if (s > target)//结束条件
		return;
	if (s == target)//结束条件
	{
		r.push_back(p);
		return;
	}
	for (int i = start; i <= nu.size() - 1; i++)
	{
		if (i>0&&nu[i-1] == nu[i] && u[i-1] == false)//在前一个树枝使用过nu[i-1]
		{
			continue;
		}
		u[i] = true;//标志已经使用过
		s = s + nu[i];
		p.push_back(nu[i]);//处理节点

		BackTrack(i+1,target,s,nu, p, u, r);
		u[i] = false;
		s = s - nu[i];//回溯
		p.pop_back();//回溯

	}
}

四、完整代码

#include<iostream>
#include<vector>
#include<stdbool.h>
#include<algorithm>
using namespace std;
void BackTrack(int start,int target,int s,vector<int> nu, vector<int>& p,vector<bool>&u, vector<vector<int>>& r)
{
	if (s > target)//结束条件
		return;
	if (s == target)//结束条件
	{
		r.push_back(p);
		return;
	}
	for (int i = start; i <= nu.size() - 1; i++)
	{
		if (i>0&&nu[i-1] == nu[i] && u[i-1] == false)//也就是在同一个树层上使用过u[i-1]
		{
			continue;
		}
		u[i] = true;//标志已经使用过
		s = s + nu[i];
		p.push_back(nu[i]);//处理节点

		BackTrack(i+1,target,s,nu, p, u, r);
		u[i] = false;
		s = s - nu[i];//回溯
		p.pop_back();//回溯

	}
}
void print(vector<vector<int>> r)//打印
{
	for (int i = 0; i < r.size(); i++)
	{
		for (int j = 0; j < r[i].size(); j++)
		{
			cout << r[i][j] << " ";
		}
		cout << endl;
	}
}
int main()
{
	int n;//元素个数
	int target;//目标和
	cout << "输入元素个数：";
	cin >> n;
	vector<vector<int>>result;//存放符合条件结果的集合
	vector<int>path;//已走的路径
	vector<int>num(n);//存放元素
	vector<bool>used(n);//判断在同一个树层上是否使用过
	cout << "依次输入各元素：";
	for (int i = 0; i < n; i++)
	{
		cin >> num[i];
	}
	cout << "输入想要求解的目标和：";
	cin >> target;//目标和
	sort(num.begin(), num.end());//先将数据进行排序
	BackTrack(0,target,0,num, path, used, result);
	cout << "结果：" << endl;
	print(result);//打印
}

//输入元素个数：5
//依次输入各元素：1 1 2 2 3
//输入想要求解的目标和：4
//结果：
//1 1 2
//1 3
//2 2