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