码迷,mamicode.com
首页 > 其他好文 > 详细

深度优先搜索

时间:2014-09-27 00:38:38      阅读:245      评论:0      收藏:0      [点我收藏+]

标签:style   blog   color   io   os   ar   for   sp   div   

深度优先搜索(DFS, Depth-First Search)是搜索的手段之一。它从某个状态开始,不断地转移状态直到无法转移,然后退到前一步的状态,如此不断重复,直至找到最终的解。

部分和问题

给定整数a1、a2、……、an,判断是否可以从中选出若干个数,使它们的和恰好为k。

限制条件

  • 1 ≤ n ≤ 20
  • -10≤ ai ≤ 108
  • -10≤ k ≤ 108

输入

n = 4

a = {1,2,4,7}

k =13

输出

Yes

 

从a1开始按顺序决定每个数加或不加,在全部n个数都决定后再判断它们的和是不是k即可。因为状态数是2n+1,所以复杂度是O(2n)。代码如下:

 1 #include <iostream>
 2 #include <vector>
 3 using namespace std;
 4 
 5 int n, k;
 6 vector<int> a;
 7 
 8 //已经从前i项得到了sum,然后对于i项之后的进行分支
 9 bool dfs(int i, int sum)
10 {
11     //如果前n项都计算过了,则返回sum是否与k相等
12     if (i == n) return sum == k;
13 
14     //不加上a[i]的情况
15     if (dfs(i+1, sum)) return true;
16 
17     //加上a[i]的情况
18     if (dfs(i+1, sum+a[i])) return true;
19 
20     //无论是否加上a[i]都不能凑成k就返回false
21     return false;
22 }
23 
24 int main()
25 {
26     while (cin >> n)
27     {
28         int tmp;
29         for (int i = 0; i < n; i++)
30         {
31             cin >> tmp;
32             a.push_back(tmp);
33         }
34         cin >> k;
35         if (dfs(0, 0))
36             cout << "Yes" << endl;
37         else
38             cout << "No" << endl;
39     }
40     return 0;
41 }

 

深度优先搜索

标签:style   blog   color   io   os   ar   for   sp   div   

原文地址:http://www.cnblogs.com/bournet/p/3995653.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!