NYOJ 题目1059部分和问题(dfs)

来源:互联网 发布:yum install wget 编辑:程序博客网 时间:2024/05/16 11:09

部分和问题

时间限制: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_杨闯亮

ac代码

#include<stdio.h>#include<string.h>int n,m,ans,ok;int v[50],a[50];void dfs(int x){int i;if(ans>m)return;if(ans==m){if(!ok){ok=1;printf("YES\n");}for(i=0;i<n;i++){if(v[i])printf("%d ",a[i]);}printf("\n");}else{for(i=x;i<n;i++){if(!v[i]){v[i]=1;ans+=a[i];dfs(i+1);ans-=a[i];v[i]=0;}}}}int main(){//int n,m;while(scanf("%d%d",&n,&m)!=EOF){int i;ok=0;ans=0;memset(v,0,sizeof(v));for(i=0;i<n;i++)scanf("%d",&a[i]);dfs(0);if(!ok)printf("NO\n");}}#include<stdio.h>#include<string.h>int n,m,ans,ok;int v[50],a[50];void dfs(int x){int i;if(ans>m)return;if(ans==m){if(!ok){ok=1;printf("YES\n");}for(i=0;i<n;i++){if(v[i])printf("%d ",a[i]);}printf("\n");}else{for(i=x;i<n;i++){if(!v[i]){v[i]=1;ans+=a[i];dfs(i+1);ans-=a[i];v[i]=0;}}}}int main(){//int n,m;while(scanf("%d%d",&n,&m)!=EOF){int i;ok=0;ans=0;memset(v,0,sizeof(v));for(i=0;i<n;i++)scanf("%d",&a[i]);dfs(0);if(!ok)printf("NO\n");}}


0 0
原创粉丝点击