Challenge-2.1.4 部分和問題

標籤：challenge

部分和問題

描述

給定整數a1、a2、.......an，判斷是否可以從中選出若干數，使它們的和恰好為K。

輸入

首先，輸入n，表示數的個數。
接著一行n個數。

（1<=n<=20,保證不超int範圍）

下一行輸入k，表示數的和

輸出

如果和恰好可以為k，輸出“YES”，並按輸入順序依次輸出是由哪幾個數的和組成，否則“NO”

範例輸入

4 1 2 4 7

13

範例輸出

YES

簡單的dfs

#include<iostream>
using namespace std;
#define MAX_N 30 
int a[MAX_N];
int n,k;


bool dfs(int i,int sum)//已經從前i項得到了和sum，然後對於 i 項之後的進行分支 
{
if(i==n)return sum==k;//如果前n項都計算過了，則返回sum是否與k相等 
if(dfs(i+1,sum))return true;//不加a[i] 
if(dfs(i+1,sum+a[i]))return true;//加a[i] 
return false;//是否加上a[i]都不能湊成k 
} 


int main()
{
cin>>n;
for(int i=0;i<n;i++)
{
cin>>a[i];
}
cin>>k;
if(dfs(0,0))
cout<<"YES"<<endl;
else 
cout<<"NO"<<endl;
return 0;
}

Challenge-2.1.4 部分和問題