NYOJ 部分和问题(入门题)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/chimchim04/article/details/78680028

部分和问题(入门题)

题目描述

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


展开阅读全文

没有更多推荐了,返回首页