部分和问题(DFS)

2020-09-14 11:27:01 浏览数 (1)

题意:给定整数a1,a2,a3,a4,a5,an。判断是否可以从中选出若干数,使它们的和恰好为k。

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

代码语言:javascript复制
#include<bits/stdc  .h>
#define maxn 20006
using namespace std;

int a[maxn];

void dfs(int i,int sum){
	if(i == n) return sum == k;
	
	if(dfs(i 1,sum)) return true;//不加第i个数 
	
	if(dfs(i 1,sum a[i])) return true;//加上第i个数 
	
	return false;
}

void solve(){
	if(dfs(0,0)) cout<<"Yes"<<endl;
	else cout<<"No"<<endl;
}

int main(){
	int n,k;
	cin>>n>>k;
	for(int i=0;i<n;i  ) cin>>a[i];
	
	solve();
	
	return 0;
}
dfs

0 人点赞