51Nod 1268 和为K的组合

51Nod  1268  和为K的组合

1268 和为K的组合

基准时间限制：1 秒 空间限制：131072 KB 分值: 20 难度：3级算法题

给出N个正整数组成的数组A，求能否从中选出若干个，使他们的和为K。如果可以，输出："Yes"，否则输出"No"。

Input

第1行：2个数N, K, N为数组的长度, K为需要判断的和(2 <= N <= 20，1 <= K <= 10^9)
第2 - N + 1行：每行1个数，对应数组的元素A[i] (1 <= A[i] <= 10^6)

Output

如果可以，输出："Yes"，否则输出"No"。

Input示例

5 13
2
4
6
8
10

Output示例

No

题解：

使用dfs暴搜！

#include <iostream>
#include <cstdio>
using namespace std;
const int maxn = 22; 

int n, flag;
long long m, sum, num[maxn], s[maxn]; 

void dfs(int pt, long long t){
	if(flag){ return; }
	if(t == m){
		flag = 1;
		return;
	}
	if(t > m){
		return;
	}
	if(t + (s[n-1]-s[pt]) < m){
		return;
	}
	if(pt+1 < n){
		dfs(pt+1, t+num[pt+1]);
		dfs(pt+1, t);
	}
}

int main(){
	freopen("in.txt", "r", stdin); 

	while(scanf("%d %lld", &n, &m) != EOF){
		sum = 0;
		for(int i=0; i<n; ++i) {
			scanf("%lld", &num[i]);
			sum += num[i];
			if(i == 0){
				s[0] = num[i];
			} else {
				s[i] = s[i-1] + num[i];
			}
		}
		if(sum < m){
			printf("No\n");
		}else{
			flag = 0;
			dfs(0, num[0]);
			dfs(0, 0);
			if(flag){
				printf("Yes\n");
			}else{
				printf("No\n");
			}
		}
	}
	return 0;
}