NYOJ 1058 部分和问题

来源：互联网发布：mac限免软件编辑：程序博客网时间：2024/05/16 12:15

部分和问题

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

难度：2

描述

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

输入

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

输出

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

样例输入

4 131 2 4 7

样例输出

YES2 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 0x3f3f3f3f3fusing 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");        }    }}

1 0