NYOJ-1058 DFS部分求和

最新推荐文章于 2019-05-03 00:51:29 发布

weixin_30821731

最新推荐文章于 2019-05-03 00:51:29 发布

阅读量117

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/bianjunting/p/10454749.html

本文介绍了一个使用暴力深度优先搜索（DFS）解决子集求和问题的算法实现。通过递归地检查每个元素是否加入当前子集，以达到目标和k。若找到符合条件的子集，则将其存入栈并输出。

原题链接

本题思路：暴力深搜，具体见代码。

 1 #include <iostream>
 2 #include <stack>
 3 using namespace std;
 4 
 5 const int maxn = 20 + 5;
 6 int a[maxn];
 7 int n, k;
 8 bool flag;
 9 stack <int > s;
10 
11 bool dfs(int i, int sum) {
12     if(i == n) return sum == k;//当搜到最后一个数时返回sum == k
13     if(dfs(i + 1, sum))  return true;//不加这个数的情况
14     if(dfs(i + 1, sum + a[i])) {//如果添加这个数满足k == sum 则将这个数加入栈中
15         s.push(a[i]);
16         return true;
17     }
18     return false;
19 }
20 
21 int main () {
22     while(cin >> n >> k) {
23         flag = true;
24         if(n == 0)  break;
25         for(int i = 0; i < n; i ++)
26             cin >> a[i];
27         flag = dfs(0, 0);
28         if(flag) {
29             cout << "YES" << endl;
30             while(!s.empty()) {
31                 cout << s.top() << " ";
32                 s.pop();
33             }
34         } else cout << "NO" << endl;
35     }
36     return 0;
37 }