部分和问题——一个简单的DFS

一、什么是DFS（深度优先搜索）？

      它从某个状态开始，不断地转移状态直到无法转移，然后回退到前一步的状态，继续转移到其他状态，如此不断重复，直到找到最终的解。根据深度优先搜索的特点，采用递归函数实现比较简单。

dfs基本模型：

void dfs(int step)
{
	判断边界
	if( 找到解 || 走不下去了 )
	{	...
		return; 
	} 
	继续下一步 dfs( step + 1 );  
}

一个简单的栗子（ 部分和问题 ）

描述
          给定整数a1、a2、.......an，判断是否可以从中选出若干数，使它们的和恰好为K。
输入
          n表示数的个数，

          接着一行n个数。

          k表示数的和。          
        （1<=n<=20,保证不超int范围）
输出
          如果和恰好可以为k，输出“YES”，并按输入顺序依次输出是由哪几个数的和组成，否则“NO”
样例输入
         4 

         1 2 4 7

         13

样例输出
         YES
         2 4 7

从排列组合的角度看，每个元素都有两个选择，加或者不加，那么就可以画出这一课树，因为状态数是2的n+1次方，所以复杂度是O(2的n次方)。

关于怎么构思这个bfs函数，可以这样想，

（1）目前sum加到第 i 项，那么这个函数接收两个参数，i 和 sum。

         bfs的返回值设为bool，表示执行到当前步骤时，是否满足了题意。

（2）先写结束条件，何时bfs结束？当 i==n 时结束，并且返回 sum 和 k 是否相等（ true or false ）

（3）接着，每个节点分为两种情况，向左走或向右走

         向左走符合题意，则返回 true，

         向右走符合题意，则返回 true，

 （4）以上三种条件都不符合，则返回 false。

在主函数中如何输出这几个数？

 可以自己构造一个数组 b，初始化数组元素均为0，用来标记数组a中已经加进去的对应数组下标。

最后，剪枝操作

 当sum>k时，一定不满足题意，那就可以直接return false了。

#include<iostream>
using namespace std;
int a[21],b[21];	
int n,k;

bool dfs(int i, int sum){	
	if (i==n) return sum==k;	
	else if (sum>k) return false;	
	
	if(dfs(i+1,sum)) return true;
		
	if(dfs(i+1,sum+=a[i])){
		b[i] = 1; 
		return true;
	}	
	
	return false;	
}
int main(){
  //b[22]={0};
  cin>>n;
  for(int i = 0; i < n; i++ ){
  	cin>>a[i];
  }
  cin>>k;
  if(dfs(0,0)){
  	cout<<"YES!"<<endl;
  	for(int j = 0; j < n; j++){
  		if (b[j]==1)
  		cout<<a[j]<<" ";
	  }
  }
  else cout<<"NO!";
	return 0;
}