NYOJ 1058 部分和问题（DFS）

部分和问题

时间限制： 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

#include<stdio.h> #include<string.h> int a[30],b[30]; int n,k,flag,sum; void DFS(int x) { if(sum>k) return; if(sum==k) { flag=1; printf("YES\n"); for(int i=0;i<n;i++) { if(b[i]) printf("%d ",a[i]); } printf("\n"); return; } for(int i=x;i<n;i++) { if(sum+a[i]<=k) { sum+=a[i]; b[i]=1; DFS(i+1); sum-=a[i]; b[i]=0; } } } int main() { while(scanf("%d%d",&n,&k)!=EOF) { memset(b,0,sizeof(b)); for(int i=0;i<n;i++) scanf("%d",&a[i]); flag=sum=0; DFS(0); if(!flag) printf("NO\n"); } return 0; }