和为 k 的组合(51Nod-1268)

题目

给出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;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值