题目如下
挑选数字:有一组整数,请挑选出一组数字,让他们的和等于指定的值,存在解打印,不存在打印
解决方法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;
}