DFS 部分和问题

题目:n个数的数组a[maxn],问其中是否存在某些数,他们的和为k,如果存在,返回Y,否则,返回N.

思想:从a[1]开始决定要不要加到sum上,如果到最后一个数也判断完成了,那么只需判断sum和k是否相等即可,用DFS来实现(递归)

示例代码:

#include <iostream>
using namespace std;
const int maxn = 1000;
int a[maxn];
int n,k;
void GetData()
{
	cin>>n;
	for(int i = 0;i < n;i++)
	{
		cin>>a[i];
	}
	cin>>k;
}
bool dfs(int i,int sum)
{
	if(i == n) return true;
	if(dfs(i+1,sum)) return true;//不加a[i] 
	if(dfs(i+1,sum+a[i])) return true;//加a[i]
	else return false;
}
void solve()
{
	if(dfs(0,0)) cout<<"Y"<<endl;
	else cout<<"N"<<endl;
}
int main()
{
	GetData();
	solve();
	return 0;
}

 

展开阅读全文

没有更多推荐了,返回首页