题目:
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搜索,两种情况,加上当前的数或者不加这个数。
AC代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n,k,flag;
int SUM;
int a[30];
void dfs(int cnt,int sum)
{
if(sum==k||(SUM-sum)==k)//注意这个if语句要放在最前面
{
flag=1;
return;
}
if(flag) return;
if(sum>k) return;
if(cnt>n)
{
// flag=0;
return;
}
dfs(cnt+1,sum+a[cnt]);
dfs(cnt+1,sum);
}
int main()
{
// int n,k;
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
SUM+=a[i];
}
dfs(1,0);
if(flag) cout<<"Yes"<<endl;
else cout<<"No"<<endl;
return 0;
}