部分和问题

试题描述

给定整数a1、a2、.......an，判断是否可以从中选出若干数，使它们的和恰好为K。

输入

第一行为两个整数n和k，n表示数的个数，k表示数的和。接下来一行为n个数a1、a2、...a。

输出

如果和恰好可以为k，输出“YES”，并按输入顺序依次输出是由哪几个数的和组成，否则输出“NO”

输入示例

4 13
1 2 4 7

输出示例

YES
2 4 7

其他说明

（1<=n<=20,保证不超int范围）

#include <iostream>

using namespace std;
int a[100001],b[100001];
int n,k,cnt=0;
int dfs(int i,int sum)
{
    if(i==n) return sum==k;  //先判断当前的sum是否和k相等 
    //开始分类讨论 
    if(dfs(i+1,sum)==1) return 1;  //不加上a[i+1] 
    if(dfs(i+1,sum+a[i])==1) {cnt++;b[cnt]=a[i];return 1;} //加上a[i+1] 
    return 0; //无论加不加上都不等于k 
}
int main()
{
    scanf("%d%d",&n,&k);
    for(int i=0;i<n;i++) scanf("%d",&a[i]);
    if(dfs(0,0)==1)   //从最开始递归，一直在内部自己调用自己，最终返回一个值 
    {
        printf("YES");    
        printf("
");    
        printf("%d",b[cnt]); //格式要求 
        for(int i=cnt-1;i>=1;i--) cout<<" "<<b[i];
    }
    else printf("NO");
    //system("pause");
    return 0;
}

经典递归