http://lx.lanqiao.cn/problem.page?gpid=T290
算法训练 和为T
时间限制:1.0s 内存限制:256.0MB
问题描述
从一个大小为n的整数集中选取一些元素,使得它们的和等于给定的值T。每个元素限选一次,不能一个都不选。
输入格式
第一行一个正整数n,表示整数集内元素的个数。
第二行n个整数,用空格隔开。
第三行一个整数T,表示要达到的和。
输出格式
输出有若干行,每行输出一组解,即所选取的数字,按照输入中的顺序排列。
若有多组解,优先输出不包含第n个整数的;若都包含或都不包含,优先输出不包含第n-1个整数的,依次类推。
最后一行输出总方案数。
样例输入
5
-7 -3 -2 5 9
0
样例输出
-3 -2 5
-7 -2 9
2
数据规模和约定
1<=n<=22
T<=maxlongint
集合中任意元素的和都不超过long的范围
注意变量的重复问题,可能不会报错
#include <cstdio>
#include <cstring>
#include <iostream>
#include <cmath>
#include <stack>
#include <queue>
#include <algorithm>
using namespace std;
int ans,n;
int a[25];
int book[25];
int res=0;
void dfs(int step,int sum,int p)
{
if(step==0&&sum!=ans)
{
return ;
}
if(step==0&&p==0)
{
return ;
}
if(step==0&&sum==ans&&p!=0)
{
int flag=0;
for(int i=1;i<=n;i++)
{
if(book[i]==1)
{
if(flag==1)
{
printf(" ");
}
printf("%d",a[i]);
flag=1;
}
}
printf("\n");
res++;
return ;
}
for(book[step]=0;book[step]<=1;book[step]++)
{
if(book[step]==1)
{
dfs(step-1,sum+a[step],p+1);
}
else
{
dfs(step-1,sum,p);
}
}
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
}
scanf("%d",&ans);
dfs(n,0,0);
printf("%d\n",res);
return 0;
}