部分和问题

题目描述

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

输入

有多组测试数据。每组测试数据两行：

第一行：正整数n、整数k，n表示数的个数，k表示数的和。

第二行：n个数

输出

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

样例输入

5 20
10 2 4 7 1
4 10
1 2 -10 4

样例输出

YES
10 2 7 1
NO

题目分析：首先做这一题的时候不难想到用穷举的思想来做题。但是，关键是怎样穷举呢？其实每一个数也就两种情况，加或者不加。emmm，为了方便理解怎样穷举，我想画一个二叉树方便大家理解。（1表示加，0表示不加）
（以第二例为例）

解法一：
画出二叉树来之后可以发现这题其实可以用深度优先搜索算法，看能不能找出符合条件的情况。下面就直接放代码了（冷静分析.jpg）

#include <bits/stdc++.h>
using namespace std;
int k,flag[100],have_find,a[100],n,counter;//flag标记是加还是不加，counter是为了输出时最后一个数据不加空格用的
bool dfs(int i,int sum)
{
    if(i==n)
        return sum==k;
    if(dfs(i+1,sum+a[i]))//加
    {
        flag[i]=1;
        return true;
    }
    if(dfs(i+1,sum))//不加
    {
        flag[i]=0;
        return true;
    }
    return false;
}
int main() {
    int i;
    while(cin>>n) {
        have_find=0,counter=0;
        cin>>k;
        for(i=0; i<n; i++)
            cin>>a[i];
        if(dfs(0,0)) {
            int j=0;
            for(i=0; i<n; i++) {
                if(flag[i])
                    counter++;
            }
            cout<<"YES"<<endl;
            for(i=0; i<n; i++) {
                if(flag[i]) {
                    j++;
                    if(j<counter)
                        cout<<a[i]<<" ";
                    else
                        cout<<a[i];
                }
            }

            cout<<endl;
        } else
            cout<<"NO"<<endl;
    }
}

解法二：根据罗列的二叉图可以发现，最多罗列了2的n次方次，如果找到解则输出。这时，我们可以根据二叉树那样，用1表示加，用0表示不加。

#include<bits/stdc++.h>
using namespace std;
int result[100];//储存要加的数字
int getset(int n,int k,int array[])
{
    int sum,j,i,counter,loc;
    for(i=pow(2,n)-1;i>=0;i--)
    {
        j=i;
        counter=0;//记录要加的个数
        loc=0;//从第一个数开始判断是否要加
        sum=0;
        while(1)
        {
            if(!j)
                break;
            if(j%2==1)
            {
                result[counter++]=array[loc];
                sum+=array[loc];
            }
            loc++;//下一个数
            j/=2;
            if(sum==k)
                return counter;
        }
    }
    return 0;
}

int main()
{
    int n,k;
    while(cin>>n)
    {
        cin>>k;
        int i,array[100],counter=0;
        for(i=0;i<n;i++)
        {
            cin>>array[i];
        }
        counter=getset(n,k,array);
        if(counter>0)
        {
            cout<<"YES"<<endl;
            for(i=0;i<counter;i++)
            {
                if(i<counter-1)
                    cout<<result[i]<<" ";
                else
                    cout<<result[i]<<endl;
            }
        }
        else
            cout<<"NO"<<endl;
    }
}

现在是不是感觉这题很简单了呢？哈哈，由于作者水平有限，如果有什么没写对，请联系作者：2596736318（qq和微信都是）