部分和问题
时间限制: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
#include <iostream>
#include <cstring>
using namespace std;int a[25],visit[25],sum,sum1,n,flag;
void DFS(int location)
{
if(flag==1) //找到之后按层依次返回
return;
if(sum<=sum1)
{
if(sum==sum1)
{
flag=1;
cout<<"YES"<<endl;
for(int i=0;i<n;i++)
if(visit[i]==1) //选中标记的输出
cout<<a[i]<<" ";
cout<<endl;
}
return;
}
for(int i=location;i<n;i++)
{
sum1+=a[i]; //选中第i个元素
visit[i]=1; //标记第i个元素
DFS(i+1); //向下搜索
sum1-=a[i]; //不选第i个元素
visit[i]=0; //解除标记
}
}int main()
{
while(cin>>n>>sum)
{
sum1=0;
memset(visit,0,sizeof(visit));
for(int i=0;i<n;i++)
cin>>a[i];
flag=0;
DFS(0);
if(!flag)
cout<<"NO"<<endl;
}
return 0;
}
原文地址:https://www.cnblogs.com/tianzeng/p/8535405.html
时间: 2024-10-31 10:41:10