NYOJ 1058 部分和问题 (DFS)

部分和问题

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

来源

经典题目

上传者

TC_杨闯亮

可能有多组解，只输出一直=组

#include<stdio.h>#include<string.h>#include<algorithm>using namespace std;int vis[110],n,k,flag,a[110],num;void dfs(int x){if(num>=k){if(num==k){if(!flag){flag=1;printf("YES\n");}for(int i=0;i<n;i++)if(vis[i])printf("%d ",a[i]);printf("\n");}return ;}for(int i=x;i<n;i++){num+=a[i];vis[i]=1;dfs(i+1);vis[i]=0;num-=a[i];}}int main(){while(scanf("%d%d",&n,&k)!=EOF){for(int i=0;i<n;i++)scanf("%d",&a[i]);memset(vis,0,sizeof(vis));flag=0;num=0;dfs(0);if(flag==0)printf("NO\n");}return 0;}