问题描述
从一个大小为n的整数集中选取一些元素,使得它们的和等于给定的值T。每个元素限选一次,不能一个都不选。
输入格式
第一行一个正整数n,表示整数集内元素的个数。
第二行n个整数,用空格隔开。
第三行一个整数T,表示要达到的和。
第二行n个整数,用空格隔开。
第三行一个整数T,表示要达到的和。
输出格式
输出有若干行,每行输出一组解,即所选取的数字,按照输入中的顺序排列。
若有多组解,优先输出不包含第n个整数的;若都包含或都不包含,优先输出不包含第n-1个整数的,依次类推。
最后一行输出总方案数。
若有多组解,优先输出不包含第n个整数的;若都包含或都不包含,优先输出不包含第n-1个整数的,依次类推。
最后一行输出总方案数。
样例输入
5
-7 -3 -2 5 9
0
-7 -3 -2 5 9
0
样例输出
-3 -2 5
-7 -2 9
2
-7 -2 9
2
数据规模和约定
1<=n<=22
T<=maxlongint
集合中任意元素的和都不超过long的范围
这个题就是简单的回溯法,每个数要么选要么不选,解的结构就是一颗二叉树,由于n<=22,时间复杂性也不高。
#include <iostream>
using namespace std;
int visited[22];//用来记录节点是否被访问过
int seq[22];//用来盛放序列
int n,ans,cnt=0;//n表元素个数,ans是结果,cnt是结果的个数
void backtrack(int t)//因为题目要求先输出不包含第n个元素的所以数组要倒着遍历
{
if(t<0)
{
int sum=0,sumofv=0;
for(int i=0;i<n;i++)
{
if(visited[i])
sum+=seq[i];
sumofv+=visited[i];//用来看是否所有节点都未被访问过
}
if(sum==ans&&sumofv)//如果序列中有节点被访问过,并且和是题目要求的就输出
{
for(int i=0;i<n;i++)
if(visited[i])
cout<<seq[i]<<" ";
cout<<endl;
cnt++;//解的个数要++
}
return;
}
for(int i=0;i<=1;i++)
{
visited[t]=i;
backtrack(t-1);
}
}
int main()
{
cin>>n;
for(int i=0;i<n;i++)
cin>>seq[i];
cin>>ans;
backtrack(n-1);
cout<<cnt<<endl;
return 0;
}