部分和问题
时间限制: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
算法思想: 回溯
使用方式: DFS 的递归形式
关于回溯 ,DFS ,递归 的解释可以参考我另一篇博客素数环
我的代码:
#include<iostream>
#include<cstdio>
#include<cstring>
int n,k;
int a[30];
bool visited[30];
int sum=0;
int flag=0;
void DFS(int i)//从i开始选,但是i还没有被选
{
//递归形式的第一部分:递归在简单情境下的退出
if(i>n || flag==1)
return;
//递归形式的第二部分:由于此时还不是简单情形,那么就要继续将改问题减少成更小的子问题
for(int j=i;j<=n;j++)
{
visited[j]=true;
if(k==sum+a[j] && flag==0 )
{
flag=1;
visited[j]=true;
printf("YES\n");
for(int p=1;p<=n;p++)
{
if(visited[p]==true)
printf("%d ",a[p]);
}
printf("\n");
return;
}
else if(k!=sum+a[j] && flag==0 )
{
sum+=a[j];
visited[j+1]=true;
DFS(j+1);
sum-=a[j];
visited[j+1]=false;
}
visited[j]=false;
}
return;
}
int main()
{
while(scanf("%d%d",&n,&k)!=EOF)
{
for(int i=1;i<=n;i++)
{
scanf("%d",a+i);
}
memset(visited,false,sizeof(visited));
sum=0;
flag=0;
DFS(1);
if(flag==0)
printf("NO\n");
}
return 0;
}