ACM部分和问题

scu_fanke

于 2017-03-29 12:39:36 发布

阅读量2k

点赞数 3

文章标签： ACM 暴力数组

本文链接：https://blog.csdn.net/fanke666/article/details/68063260

版权

部分和问题

时间限制： 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的定义解决，不过最让人头痛的是如何实现剪枝，减少不必要的时间浪费，剪枝还不熟练，这道题需要减掉的是

1.从当前状态如何转移都不会存在解

2.当sum超过k时，也没必要继续搜索

#include<cstdio>
#include<iostream>
using namespace std;
int n,k,a[22],b[22];
 bool  dfs(int x,int sum)  //从左到右遍历一遍可得解
  if(sum>k) return false;    //此处属于后期补充，当代码加上这一个剪枝后，时间就会变为8MS，所以平时还要多注重剪枝，考虑优化代码啊！！！
   if(x==n) 
   return sum==k;  //如果前n项计算过了，返回sum=k是否相等
   if(dfs(x+1,sum)){
   b[x]=0;
   return true;
   }   //如果不加上a[x]的情况，标记为0；
   if(dfs(x+1,sum+a[x])){
   b[x]=1;
   return true;
   }   //如果加上a[x]的情况，标记为1；
   return false; 
 }
 int main(){
   cin>>n;
for(int i = 0;i<n; i++)
scanf("%d",&a[i]);
cin>>k;
if(dfs(0,0)){
printf("YES\n");
for(int i=0;i<n;i++)
 if(b[i])printf("%d ",a[i]);
printf("\n");
     }
}