部分和问题 南阳acm1058（递归+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

来源

经典题目

上传者

TC_杨闯亮

简单的搜索问题，要注意它有多组测试数据

 #include<stdio.h>
 #include<string.h>
 int a[],b[];
 int n,k,flag;
 void dfs(int x,int sum)
 {
     int i;
     if(sum > k) return;    //此时不需要再相加了，已经超了
     if(x==n)
     {
         if(sum == k)
         {
             flag = ;
             printf("YES\n");
             for(i=; i<n; i++)
             {
                 if(b[i]) printf("%d ",a[i]);
             }
             printf("\n");
             return;
         }
         return;
     }
     b[x] = ;
     dfs(x+,sum);    //sum不需要+a[x]时，b[x] = 0
     b[x] = ;
     dfs(x+,sum+a[x]);    //sum+a[x]时，b[x] = 1
 }
 int main()
 {
     int i;
     while(scanf("%d%d",&n,&k)!=EOF)
     {
             flag = ;
             memset(b,,sizeof(b));
         for(i=; i<n; i++)
             scanf("%d",&a[i]);
         dfs(,);
         if(!flag)
             printf("NO\n");
     }
     return ;
 }