《挑战程序竞赛》 2.1.4 部分和问题

Posted 2020-09-21

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

 

解法：利用dfs深度优先遍历，从a1开始按顺序决定每个数是加还是不加。

 

code：

 1 #include <iostream>
 2 #include <algorithm>
 3 using namespace std;
 4 
 5 #define maxn 100000
 6 
 7 int k, n;
 8 
 9 int a[maxn];
10 
11 bool dfs(int i, int sum)
12 {
13     if(i == n)
14         return sum == k;
15 
16     if(dfs(i + 1, sum + a[i]))
17         return true;
18 
19     if(dfs(i + 1, sum))
20         return true;
21 
22     return false;
23 
24 }
25 
26 int main()
27 {
28 
29     while(cin>>n)
30     {
31         for(int i=0; i<n; i++)
32             cin>>a[i];
33         cin>>k;
34         if(dfs(0, 0))
35             cout<<"Yes"<<endl;
36         else
37             cout<<"No"<<endl;
38 
39     }
40     return 0;
41 }