《挑戰程序競賽》 2.1.4 部分和問題
阿新 • • 發佈:2017-06-03
als 按順序 space code log pre std mes namespace
題意:給定整數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 returnsum == 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 }
《挑戰程序競賽》 2.1.4 部分和問題