nyoj 1058部分和问题

部分和问题

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