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