nyoj 部分和问题（深度优先搜索DFS）

部分和问题(入门题)

描述

给你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<iostream>
#include<stack>
using namespace std;
int a[25],n,k;
int dfs(int i,int sum)
{
    if(sum>k)//进行剪枝
        return 0;
    if(i==n)
        return sum==k?1:0;//当i等于n时，返回sum是否等于k
    if(dfs(i+1,sum))//不加a[i]
        return 1;
    if(dfs(i+1,sum+a[i]))//加上a[i]
        return 1;
    return 0;
}
int main()
{
    while(cin>>n>>k)
    {
        int i;
        for(i=0;i<n;i++)
            cin>>a[i];
        if(dfs(0,0))
            cout<<"YES"<<endl;
        else
            cout<<"NO"<<endl;
    }
    return 0;
}

                                                                        部分和问题

描述

给定整数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<iostream>
#include<stack>
using namespace std;
int a[25],n,k;
stack<int>s;
int dfs(int i,int sum)
{
    if(sum>k)//进行剪枝
        return 0;
    if(i==n)//i等于n时，开始回溯
        return sum==k;//当i等于n时，返回sum是否等于k
    if(dfs(i+1,sum))//返回不加a[i]的情况
        return 1;
    if(dfs(i+1,sum+a[i]))//返回加上a[i]的情况
    {
        s.push(a[i]);
        return 1;
    }
    return 0;//当最后最后无论如何都凑不成k时，返回0
}
int main()
{
    while(cin>>n>>k)
    {
        while(!s.empty())
            s.pop();//对栈清零
         int i;
         for(i=0;i<n;i++)
            cin>>a[i];
         if(dfs(0,0))
         {
             cout<<"YES"<<endl;
            while(!s.empty())
            {
                cout<<s.top()<<" ";
                s.pop();
            }
            cout<<endl;
         }
         else
            cout<<"NO"<<endl;
    }
    return 0;
}

      ps:深度优先搜索的入门题目，总算是明白了一点点深搜的奥妙，就从它开始打开搜索的大门 。。