nyoj 1058 部分和问题

来源:互联网 发布:巫师3和老滚5 知乎 编辑:程序博客网 时间:2024/05/19 17:11
描述
给定整数a1、a2、.......an,判断是否可以从中选出若干数,使它们的和恰好为K。
输入
首先,n和k,n表示数的个数,k表示数的和。
接着一行n个数。
(1<=n<=20,保证不超int范围)
输出
如果和恰好可以为k,输出“YES”,并按输入顺序依次输出是由哪几个数的和组成,否则“NO”
样例输入
4 131 2 4 7
样例输出
YES

2 4 7

#include<iostream>#include<cstdio>#include<cstring>using namespace std;    int n,k,f,sum;    int a[25],b[25];void dfs(int x){   int i;   if(sum>=k){    if(sum==k){     if(f){            f=0;            printf("YES\n");     }      for(i=0;i<n;i++){        if(b[i])            printf("%d ",a[i]);      }      printf("\n");   }    return ;   }    for(i=x;i<n;i++){        sum+=a[i];        b[i]=1;        dfs(i+1);        b[i]=0;        sum-=a[i];    }}int main(){    int i;    while(scanf("%d%d",&n,&k)!=EOF){        for(i=0;i<n;i++){            scanf("%d",&a[i]);        }        f=1;        sum=0;        memset(b,0,sizeof(b));        dfs(0);        if(f)            printf("NO\n");    }    return 0;}


0 0
原创粉丝点击