题目
给出N个正整数组成的数组A,求能否从中选出若干个,使他们的和为K。如果可以,输出:"Yes",否则输出"No"。
输入
第1行:2个数N, K, N为数组的长度, K为需要判断的和(2 <= N <= 20,1 <= K <= 10^9)
第2 - N + 1行:每行1个数,对应数组的元素A[i] (1 <= A[i] <= 10^6)输出
如果可以,输出:"Yes",否则输出"No"。
输入样例
5 13
2
4
6
8
10输出样例
No
思路:dfs
普通的深搜加 vis 标记会超时,因此可以对搜索顺序进行优化,每次 dfs 在当前结点的下一个结点开始搜索即可
源程序
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<utility>
#include<stack>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<bitset>
#define EPS 1e-9
#define PI acos(-1.0)
#define INF 0x3f3f3f3f
#define LL long long
const int MOD = 1E9+7;
const int N = 1000000+5;
const int dx[] = {-1,1,0,0,-1,-1,1,1};
const int dy[] = {0,0,-1,1,-1,1,-1,1};
using namespace std;
int n,k;
int a[N];
bool flag;
void dfs(int sum,int step) {
if(flag)
return;
if(sum==0) {
flag=true;
return;
}
if(step<=n){
for(int i=step;i<=n;i++){
sum-=a[i];
dfs(sum,i+1);
sum+=a[i];
}
}
}
int main() {
scanf("%d%d",&n,&k);
for(int i=1; i<=n; i++)
scanf("%d",&a[i]);
for(int i=1;i<=n;i++)
dfs(k,i);
if(flag)
printf("Yes\n");
else
printf("No\n");
return 0;
}