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

87 阅读1分钟

题目:

部分和问题(入门题)

时间限制: 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:刚开始接触搜索,理解的不很透彻,先用道入门题理解理解