问题链接:http://acm.nyist.net/JudgeOnline/problem.php?pid=1058
部分和问题
时间限制: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
分析:
一共输入n个数,每个数有两种状态,分别为加入解集合、不加入解集合。
每个数作为解空间树的一层,则树有n层。利用递归调用对树进行深度优先遍历,即可找到解。
因为n<=20,所以树空间大小最大为2^20,在百万级别以内。
代码:
#include <iostream>
#include <stdio.h>
#define MAX 21
/* run this program using the console pauser or add your own getch, system("pause") or input loop */
int num[MAX];
bool mark[MAX];
int n,k;
bool dfs(int i,int sum){
//递归遍历到第i个数
if(i == n) { //每次都深度递归到i == n 然后判断进行返回
return sum == k;
}
//加上第i个数
mark[i] = false;
if(dfs(i+1,sum)){
return true;
}
//不加第i个数
mark[i] = true;
if(dfs(i+1,sum+num[i])){
return true;
}
//加上和不加最终都得不到sum == k 则返回false
return false;
}
int main(int argc, char** argv) {
while(scanf("%d%d",&n,&k)!=EOF){
//1.输入n个数
for(int i=0;i<n;i++){
scanf("%d",&num[i]);
mark[i] = false;
}
//2.深度搜索dfs(i,j)表示遍历到第i个数,此时的和为j
if(dfs(0,0)){
printf("YES\n") ;
for(int i=0;i<n;i++){
if(mark[i] == true){
printf("%d ",num[i]) ;
}
}
printf("\n");
}else{
printf("NO\n");
}
}
return 0;
}