NYOJ 部分和问题(入门题)

部分和问题(入门题)

题目描述

给你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

提示

无

来源

nyoj

就是简单深搜，原谅我身为理科生实在是不知道怎么写注释，就将就着看吧！！

代:1：

#include<stdio.h>
#include<algorithm>
using namespace std;
int n,k,a[105],i;
bool dfs(int i,int sum)
{
    if(i==n) return sum==k; 
    if(dfs(i+1,sum)) return 1;
    if(dfs(i+1,sum+a[i])) return 1;
    return 0;
}
int main()
{
    while(~scanf("%d%d",&n,&k))
    {
        for(int i=0;i<n;i++)
            scanf("%d",&a[i]);
        printf("%s\n",dfs(0,0)?"YES":"NO");   //如果sum==k输出YES，否则输出NO
    }
    return 0;
}
代码2：（其实就只加了一个判断条件，其他都一样的，但是剪枝之后时间会缩短很多）

#include<stdio.h>
int a[50],n,k;
int dfs(int i,int sum)
{
    if(sum>k) return 0;
    if(i==n) return sum==k;
    if(dfs(i+1,sum+a[i])) return 1;
    if(dfs(i+1,sum)) return 1;
    return 0;
}
int main()
{
    while(~scanf("%d%d",&n,&k))
    {
         for(int i=0;i<n;i++)
        scanf("%d",&a[i]);
    printf("%s\n",dfs(0,0)?"YES":"NO");
    }
    return 0;
}