nyoj1058部分和问题（简单基础易上手的dfs）

最新推荐文章于 2018-07-18 20:35:28 发布

iwi_ac

最新推荐文章于 2018-07-18 20:35:28 发布

阅读量613

点赞数

分类专栏：数据结构算法基础

本文链接：https://blog.csdn.net/u012349696/article/details/42834107

版权

算法基础同时被 2 个专栏收录

64 篇文章 0 订阅

订阅专栏

数据结构

17 篇文章 0 订阅

订阅专栏

部分和问题

时间限制： 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

来源

经典题目

上传者

TC_杨闯亮

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;
int a[50];
int vis[50];
int n,k,flag;
int sum;
void dfs(int pos)
{
    if(flag)
        return ;
    if(sum>=k)
    {
        if(sum==k)
        {
            flag=1;
            printf("YES\n");
            for(int i=0;i<n;i++)
                if(vis[i])
                printf("%d ",a[i]);
                printf("\n");
        }
        return ;
    }
    for(int i=pos;i<n;i++)
    {
        sum+=a[i];
        vis[i]=1;
        dfs(i+1);
        sum-=a[i];
        vis[i]=0;
    }
}
int main()
{
    while(~scanf("%d%d",&n,&k))
    {
        memset(vis,0,sizeof(vis));
        flag=0;
        for(int i=0;i<n;i++)
            scanf("%d",&a[i]);
        dfs(0);
        if(flag==0)
            printf("NO\n");
    }
}