414-回溯算法-挑数字问题(回溯高效)

题目如下

挑选数字：有一组整数，请挑选出一组数字，让他们的和等于指定的值，存在解打印，不存在打印

解决方法1（子集树+剪枝）

r是未处理的元素的和，未处理的节点是节点已经访问处理过了，不管是选择了还是不选择，就是算处理过了

int arr[] = { 4,8,12,16,7,9,3 };
const int length = sizeof(arr) / sizeof(arr[0]);
int number = 18;
vector<int> x;//记录选择的数字
int sum = 0;//记录所选择数字的和
int r = 0;//记录未处理的数字的和
void func(int i)
{
	if (i == length)//到达叶子节点
	{
		if (sum != number)
		{
			return;
		}
		for (int v : x)
		{
			cout << v << " ";
		}
		cout << endl;
	}
	else//深度遍历的过程中
	{
		r -= arr[i];//处理当前i节点
		//r是未处理的元素的和，未处理的节点是节点已经访问处理过了，不管是选择了还是不选择，就是算处理过了
		if (sum + arr[i] <= number)//剪左树枝  已选择的数字的和+即将要选择的数字
		{
			sum += arr[i];
			x.push_back(arr[i]);
			func(i + 1);
			sum -= arr[i];
			x.pop_back();
		}

		if (sum + r >= number)//剪右树枝  已选择的数字的和+剩余的可以被选择的数字的和(i+1,i+2....n)>=number才进入右孩子
		{
			func(i + 1);
		}
		r += arr[i];//回溯到父节点
	}
}
int main()
{
	for (int v : arr)
	{
		r += v;
	}
	func(0);
	return 0;
}

解决方法2（更高效的回溯思想）

枚举法
先举个例子：
我们在4, 5，6，7中挑选数字，和 满足：
满足条件的number：18

我们可以挑以4开头，也可以挑以5开头，也可以挑以6开头，也可以挑以7开头。

形成了1棵四叉树。
往左边走，深度遍历，如果先选择4，选了4以后，还能选什么呢？就是看4后边有3个元素（5,6，7），还可以选择5或者6或者7。然后选了5之后，还剩下6,7，然后选了6，还有7。

如果先选择5，就不选4了，因为选4的时候已经把5选过了。以此类推。

一共有8个分枝到达叶子节点。（8个叶子节点）
如果用子集树就是2^4=16，用于处理目标不明确的。
而此题目标是明确的，求选择的数字的和指定值。

我们拿着18遍历去减，判断。

根节点往左走，选了4,18-4,14，然后选择5，14-5=9，然后选择6,9-6=3，然后底下的7大于3，所以这路找不出来，回退到父节点5，判断7，以此类推下去。
递归1次就是1叉树，递归2次就是而2叉树，递归3次就是3叉树…

4次for循环，形成四叉树

循环的意思是：从当前元素开始一直遍历到后面的剩余的所有元素进行递归，递归几次就是几叉树，以此类推。number==0，代表找到一组序列了

int arr[] = { 4,8,12,16,7,9,3,3 };
const int length = sizeof(arr) / sizeof(arr[0]);
int number = 18;
vector<int> vec;//存放选择的数字
void func(int i, int number)
{
	if (number == 0)
	{
		for (int v : vec)
		{
			cout << v << " ";
		}
		cout << endl;
	}
	else
	{
		//以当前节点开始，把剩余元素的孩子节点生成
		for (int k = i; k < length; ++k)
		{
			if (number >= arr[k])//arr[K]值小于number（待组成的元素值），才可以选择arr[k]，不然number减一下就变成负数了
			{
				vec.push_back(arr[k]);
				//当前层遍历到下一层的时候   arr[k] != arr[k-1]
				//如果不允许重复选择元素：
				func(k + 1, number - arr[k]);
				//k+1就是少一个元素，孩子节点少1个叉树
				//number扣除当前选择的值
				//遍历孩子节点，arr[k]的孩子节点
				
				//如果允许重复选择元素：
				//func(k, number - arr[k]);
				//孩子的叉树数和父亲的叉树数一样
				//相当于当前元素可以重复选择
				
				vec.pop_back();//回溯到父节点
			}
		}
	}
}
int main()
{
	func(0, number);
	return 0;
}