NYOJ1058+NYOJ1282 部分和问题(深度优先搜索DFS入门题)

题目:

部分和问题(入门题)

时间限制:1000 ms  |  内存限制:65535 KB
难度:0
描述

给你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
代码:

#include <stdio.h>
#include <string.h>
#include <math.h>
#include <stack>
#include <algorithm>
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
int n,i,m,j,k;
int a[50];
bool dfs(int i,int sum)
{
    if(i==n)return sum==k;//返回条件    
    if(dfs(i+1,sum))return true;//不加a[i]的情况
    if(dfs(i+1,sum+a[i]))return true;//加上a[i]的情况
    return false;//不论有没有返回a[i],没有凑成k就返回false
}
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;
}



题目:

部分和问题

时间限制:1000 ms  |  内存限制:65535 KB
难度:2
描述
给定整数a1、a2、.......an,判断是否可以从中选出若干数,使它们的和恰好为K。
输入
首先,n和k,n表示数的个数,k表示数的和。
接着一行n个数。
(1<=n<=20,保证不超int范围)
输出
如果和恰好可以为k,输出“YES”,并按输入顺序依次输出是由哪几个数的和组成,否则“NO”
样例输入
4 13
1 2 4 7
样例输出
YES
2 4 7
借一张图来粗浅理解一下:


代码:

#include <stdio.h>
#include <string.h>
#include <math.h>
#include <stack>
#include <algorithm>
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
int n,i,m,j,k;
int a[50];
stack<int>p;
bool dfs(int i,int sum)
{
    if(i==n)return sum==k?1:0;//停止条件,如果前n项都计算过了,则返回sum与k的值是否相等
    if(dfs(i+1,sum))return true;//返回不加a[i]的情况
    if(dfs(i+1,sum+a[i]))//加a[i]的情况
    {
        p.push(a[i]);
        return true;
    }
    return false;//不论加不加a[i]都凑不成k就返回false
}
int main()
{
    while(~scanf("%d %d",&n,&k))
    {
        for(i=0; i<n; i++)
            scanf("%d",&a[i]);
        if(dfs(0,0))
        {
            printf("YES\n");
            while(!p.empty())//栈为非空
            {
                int x=p.top();
                printf("%d ",x);
                p.pop();
            }
            printf("\n");
        }
        else
            printf("NO\n");
    }
    return 0;
}

ps:刚开始接触搜索,理解的不很透彻,先用道入门题理解理解

发布了846 篇原创文章 · 获赞 221 · 访问量 108万+
展开阅读全文

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

©️2019 CSDN 皮肤主题: 编程工作室 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览