nyoj 1282 部分和问题(入门题)（dfs）

部分和问题(入门题)

时间限制： 1000 ms  |  内存限制： 65535 KB

难度： 0

描述

给你n个数(a1,a2,a3.......an) ,是否存在某一些数字加起来等于k,有就输出 "YES",否则输出 "NO"。

数据范围：n<20;

a1+a2+....an在int范围里面.

输入

多组输入
每组第一行输入两个数n,k
第二行输入n个数a1 a2 ...... an

输出

如果存在一些数加起来为k输出"YES";否则输出"NO".

样例输入

5 6
2 3 5 2 1
3 6
2 3 9

样例输出

YES
NO

分析：简单dfs。

AC代码：

#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=25;
int a[maxn];
int n,k;

int dfs(int i,int tol){
	if(tol>k)return 0;
	if(tol==k)return 1;
	
	int flag=0;
	for(int j=i+1;j<=n;j++){
		flag=dfs(j,tol+a[j]);
		if(flag)return 1;
	}
	return 0;
}
int main(){
	while(scanf("%d%d",&n,&k)==2){
		for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
		
		int flag=0;
		for(int i=1;i<=n;i++)
		{
			flag=dfs(i,a[i]);
			if(flag)break;
		}
		
		if(flag)printf("YES\n");
		else printf("NO\n");
	}
	return 0;
}