部分和问题(DFS)

时间:2022-07-24
本文章向大家介绍部分和问题(DFS),主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。

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

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

#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;
}