算法:搜索
题目描述
给定整数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
代码:
#include <iostream>
#include <iomanip>
#include <algorithm>
#include <cstring>
#include <string>
#include <queue>
#include <stdio.h>
using namespace std;
int a[30],b[30],n,k,flag,step,c[30],sum;
void dfs(int x)
{
if(sum==k)
{
cout<<"YES\n"<<b[0];
for(int i=1;i<step;i++)
cout<<" "<<b[i];
cout<<endl;
flag=1;
return ;
}
if(sum>k) return;
if(flag) return;
for(int j=x;j<n;j++)
{
if(!c[j])
{
c[j]=1;
b[step++]=a[j];
sum+=a[j];
dfs(j);
c[j]=0;
step--;
sum-=a[j];
}
}
}
int main()
{
int i,j,m;
while(cin>>n>>k)
{
for(i=0;i<n;i++)
cin>>a[i];
flag=0;
memset(c,0,sizeof(c));
sum=0;step=0;
dfs(0);
if(!flag) cout<<"NO"<<endl;
}
return 0;
}