您的位置:首页 > 其它

nyoj1058部分和问题(简单基础易上手的dfs)

2015-01-18 10:55 239 查看


部分和问题

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


来源经典题目
上传者TC_杨闯亮

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;
int a[50];
int vis[50];
int n,k,flag;
int sum;
void dfs(int pos)
{
if(flag)
return ;
if(sum>=k)
{
if(sum==k)
{
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=pos;i<n;i++)
{
sum+=a[i];
vis[i]=1;
dfs(i+1);
sum-=a[i];
vis[i]=0;
}
}
int main()
{
while(~scanf("%d%d",&n,&k))
{
memset(vis,0,sizeof(vis));
flag=0;
for(int i=0;i<n;i++)
scanf("%d",&a[i]);
dfs(0);
if(flag==0)
printf("NO\n");
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: