nyoj 1058部分和问题

部分和问题

时间限制：1000 ms  |  内存限制：65535 KB

难度：2

描述

给定整数a1、a2、.......an，判断是否可以从中选出若干数，使它们的和恰好为K。

输入

首先，n和k，n表示数的个数，k表示数的和。
接着一行n个数。
（1<=n<=20,保证不超int范围）

输出

如果和恰好可以为k，输出“YES”，并按输入顺序依次输出是由哪几个数的和组成，否则“NO”

样例输入

4 131 2 4 7

样例输出

YES2 4 7

来源

#include <stdio.h>  #include <string.h>  int a[22];  int mark[22];  int n,k;  int dfs(int i,int sum)  {      if(sum == k) //搜索到答案时          return 1;      if(sum > k || i == n)  //剪枝           return 0;      if(dfs(i + 1,sum))//不加的情况          return 1;      if(dfs(i + 1,sum + a[i])) //加的情况       {          mark[i] = 1;  //标记选出数的位置           return 1;      }      return 0;  //加与没加都没搜到答案的情况   }  int main()  {      while(~scanf("%d%d",&n,&k))      {          int i;          memset(mark,0,sizeof(mark));          for(i = 0;i < n;i ++ )              scanf("%d",&a[i]);          if(dfs(0,0))          {              printf("YES\n");              for(i = 0;i < n;i ++ )              if(mark[i])                  printf("%d ",a[i]);              printf("\n");          }          else              printf("NO\n");      }      return 0;  }