NYOJ 1058 部分和问题

部分和问题

时间限制： 1000 ms  |  内存限制： 65535 KB

难度： 2

描述

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

输入

首先，n和k，n表示数的个数，k表示数的和。
接着一行n个数。
（1<=n<=20,保证不超int范围）

输出

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

样例输入

4 13
1 2 4 7

样例输出

YES
2 4 7

思路：和nyoj927题基本上一致但是除了判断是否存在几个数相加等于k之外还需要如果存在的话还要输出这几个数，所以我们需要将这几个数进行标记一下，然后再用一个数组将其存起来最后将其输出即可。

代码：

#include<stdio.h>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<stdlib.h>
#include<stack>
#include<vector>
#include<string.h>
#include<map>
#define INF 0x3f3f3f3f3f
using namespace std;
int flag,sum,k,n,a[101],p,b[1001],c[1001];
void dfs(int t)
{
    int i;
    if(sum>k)
    {
        return;
    }
    if(sum==k)//如果存在几个数相加等于k就将b[i]对应的a[i]值存在c[i]中
    {
        flag=1;
        //b[p++]=a[i];
        for(i=0; i<n; i++)
        {
            if(b[i]==1)
            {
                c[p++]=a[i];
            }
        }
    }
    for(i=t; i<n; i++)
    {
        sum+=a[i];
        b[i]=1 ;//如果访问到a[i],用b[i]标记一下证明已经访问过了
        dfs(i+1);
        sum-=a[i];
        b[i]=0;
    }
}
int main()
{
    while(~scanf("%d%d",&n,&k))
    {
        flag=0;
        sum=0;
        p=0;
        for(int i=0; i<n; i++)
        {
            scanf("%d",&a[i]);
        }
        dfs(0);
        if(flag==1)
        {
            printf("YES\n");
            for(int i=0; i<p; i++)//输出c[i]
            {
                printf("%d ",c[i]);
            }
            printf("\n");
        }
        else
        {
            printf("NO\n");
        }
    }
}