部分和问题
时间限制:
1000 ms | 内存限制:
65535 KB
难度:
2
-
描述
-
给定整数a1、a2、.......an,判断是否可以从中选出若干数,使它们的和恰好为K。
-
输入
-
首先,n和k,n表示数的个数,k表示数的和。
接着一行n个数。
(1<=n<=20,保证不超int范围)
输出
- 如果和恰好可以为k,输出“YES”,并按输入顺序依次输出是由哪几个数的和组成,否则“NO” 样例输入
-
4 13 1 2 4 7
样例输出
-
YES 2 4 7
-
首先,n和k,n表示数的个数,k表示数的和。
因为可以是任意项,所以可用01背包,对于一个数字,可以选择加或者不加,直到最后,如果出现答案就跳出,直到最后没有出现就是NO
#include<stdio.h>
#include<string.h>
int m,n,a[25],b[25];
int dfs(int i,int s)
{
if(s==n&&i==m)
return 1;
if(i==m&&s!=n)
return 0;
b[i]=0;
if(dfs(i+1,s))return 1;
b[i]=1;
if(dfs(i+1,s+a[i]))return 1;
}
int main()
{
int i,j,s;
memset(b,0,sizeof(b));
while(~scanf("%d%d",&m,&n))
{
for(i=0; i<m; i++)
scanf("%d",&a[i]);
if(dfs(0,0))
{
printf("YES\n");
for(i=0; i<m; i++)
if(b[i]==1)
printf("%d ",a[i]);
printf("\n");
}
else
printf("NO\n");
}
return 0;
}