NYOJ 1058 部分和问题

/*部分和问题
描述
给定整数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 n,k,a[25],book[25];
bool DFS(int i,int sum)
{
  if(i==n)
       return sum==k;
    if(DFS(i+1，sum))
           return ture;
    if(DFS(i+1,sum+a[i]))
    {
       book[i]=1;
       return true;
     }
   return flase;
 }
  int main()
{
  while(scanf("%d%d",&n,&k))
  {
     int i;
     memset(book,0,sizeof(book));
       for(i=0;i<n;i++)
       {
          scanf("%d",&a[i]);
        }
       if(DFS(0,0))
       {
         printf("YES\n");
            for(i=0;i<n;i++)
            {
              if(book[i])
                printf("%d",a[i]);
             }
             printf("\n");
        }
        else
       printf("NO\n");
       }
        return 0;
    }

递归回溯：

#include<stdio.h>
int a[25],book[25];
int m,n,flag,sum;
void DFS(int x)
{
    int i;
    if(flag)
    return; 
    if(sum>=n)
{
    if(sum==n)
    {
        flag=1;
          printf("YES\n");
          for(i=0;i<m;++i)
          if(book[i])
          printf("%d ",a[i]);
          printf("\n");
   }
   return;
}
    for(i=x;i<m;++i)
    {
           if(!book[i])
         {
            sum=sum+a[i];
              book[i]=1;
            DFS(x+1);
              sum=sum-a[i];
            book[i]=0;      
        }   
    }
}
int main()
{   
    int i,j;
    while(~scanf("%d%d",&m,&n))
    { 
    flag=0;
      for(i=0;i<m;++i)
      {
        scanf("%d",&a[i]);
        book[i]=0;
      } 
      sum=0;
      DFS(0);
      if(!flag)
      printf("NO\n");
    }
    return 0;
}