深度优先搜索:部分和问题

#include <iostream>
#include <vector>
#include <algorithm>
#include <stack>
#define MAX_N 2147483647
using namespace std;


/*深度优先搜索
 * 搜索理解:从某个状态开始,不断地转移状态直到无法 转移,然后回退到前一步的状态,继续转移到其他状态,如此不断重复,直到找到最终的解。
*/


/* 部分和问题:给定整数a1,a2,a3,,,an,判断是否可以从中选出若干数,使它们的和恰好为k。
 * 问题解决思路:
 * 1:dfs遍历每个数列元素能达到的状态(选与不选)
 * 2:剪枝,若已经达到目标值,不需要再进行遍历其他状态,或者超过目标值时,也不需要再遍历,将其他遍历过程剪掉
 * 其他思路:定义一个全局bool值,仅当bool值为true时,将深搜停止掉,
*/
int n;  //n个数列元素
int k;  //所达到的目标值
bool my_dfs(const vector<int> &ivec, int sum, int index, bool ischoosed)
{
	if(index==ivec.size()) return false;
	if(ischoosed)
	 sum+=ivec.at(index);

	if(sum==k) return true;
	else if(sum>k) return false;
	else
	{
		++index;
		if(dfs(ivec, sum, index, true))  return true;
		if(dfs(ivec, sum, index, false)) return true;
	}

}

bool dfs(int i,int sum,const vector<int> &ivec)
{
	if(i==n) return sum==k;
	if(dfs(i+1,sum,ivec)) return true;
	if(dfs(i+1,sum+ivec.at(i),ivec)) return true;
	return false;
}

bool solve()
{
	cin>>n;
	vector<int> ivec(n,0);
	for (decltype(ivec.size()) i = 0; i !=ivec.size(); ++i) {
		cin>>ivec.at(i);
	}
	cin>>k;
	if(my_dfs(ivec, 0, 0, true) || my_dfs(ivec, 0, 0, false)) cout << "keyi" << endl;
}

int main()
{
	solve();

	return  0;
}

猜你喜欢

转载自blog.csdn.net/chongzi_daima/article/details/104092489