首页 > 代码库 > Challenge-2.1.4 部分和问题

Challenge-2.1.4 部分和问题

部分和问题

描述
给定整数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 部分和问题