nyoj 1058部分和问题

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/ACM_e/article/details/77868376


部分和问题

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


没有更多推荐了,返回首页