xynu 2917: 整数凑和问题(搜索)

2917: 整数凑和问题

时间限制: 1 Sec   内存限制: 128 MB
提交: 31   解决: 3
您该题的状态：已完成
[提交][状态][讨论版]

题目描述

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

输入

有多组数据

首先，n和k，n表示数的个数，k表示数的和。K不为0.

接着一行n个数。
（1<=n<=20,保证不超int范围）

输出

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

样例输入

4 13
1 2 4 7

样例输出

YES
2 4 7

这道题是感觉挺坑的，原以为是输出所有=k的组合，提交就wa了，后来又输出=k的第一个组合，还是wa,过了二天又写了一遍，稀里糊涂的输出符合的最后一组合，AC了。

#include<stdio.h>
#include<string.h>
using namespace std;
int mat[25];
int a[25];
int vis[25]; 
long long k,sum;//int型就可以
int n,m;
int flag;
void bfs(long long sum,int t){
    if(sum==k){
        m=t;
        flag=1;
        return;
    }
    for(int i=0;i<n;i++){
        if(!vis[i]&&sum+mat[i]<=k){
            vis[i]=1;
            a[t]=mat[i];
            bfs(sum+mat[i],t+1);
            vis[i]=0;
        }
    }
    return;
}
int main()
{
    while(~scanf("%d%lld",&n,&k)){
        flag=0;
        memset(vis,0,sizeof(vis));
        for(int i=0;i<n;i++){
            scanf("%d",&mat[i]);
        }
        bfs(0,0);
        if(flag){
            printf("YES\n");
        for(int i=m-1;i>0;i--)
            printf("%d ",a[i]);
        printf("%d\n",a[0]);
        }
        else printf("NO\n");
    }
    return 0;
}