部分和问题
时间限制: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
很明显,这是一道简单的dfs搜索,直接理由dfs的定义解决,不过最让人头痛的是如何实现 剪枝,减少不必要的时间浪费,剪枝还不熟练,这道题需要减掉的是
1.从当前状态如何转移都不会存在解
2.当sum超过k时,也没必要继续搜索
#include<stdio.h>
#include<string.h>
int a[30],book[30];
int n,j,s,sum,f;
void dfs(int step)
{
int i;
if(s>=sum )
{
if(s==sum)
{
if(f==0)
{
f=1;
}
if(f)
{
printf("YES\n");
for(i=0; i<n; i++)
if(book[i])
printf("%d ",a[i]);
printf("\n");
}
}
//return ;
}
int k;
for(k=step; k<n; k++)
{
s+=a[k];
book[k]=1;
dfs(k+1);
s-=a[k];
book[k]=0;
}
}
int main()
{
int i;
while(~scanf("%d%d",&n,&sum))
{
s=0,f=0;
memset(book,0,sizeof(book));
for(i=0; i<n; i++)
scanf("%d",&a[i]);
dfs(0);
if(f==0) printf("NO\n");
}
return 0;
}