DFS-The partial sum problem

The partial sum problem

时间限制： 1000 ms  |  内存限制： 65535 KB

难度： 2

描述

One day,Tom’s girlfriend give him an array A which contains N integers and asked him:Can you choose some integers from the N integers and the sum of them is equal to K. 

输入

There are multiple test cases.
Each test case contains three lines.The first line is an integer N(1≤N≤20),represents the array contains N integers. The second line contains N integers,the ith integer represents A[i](-10^8≤A[i]≤10^8).The third line contains an integer K(-10^8≤K≤10^8).

输出

If Tom can choose some integers from the array and their them is K,printf ”Of course,I can!”; other printf ”Sorry,I can’t!”.

样例输入

4
1 2 4 7
13
4
1 2 4 7
15

样例输出

Of course,I can!
Sorry,I can't!

题目大意就是给你几个数，然后给你一个结果，这几个数组合加起来能不能等于这个数。

dfs水过，有大神剪枝，花了12MS，我花了884MS。。。

我的AC代码

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
int a[22], book[22], N, K;
bool dfs(int step, int sum) {
	if (step == N) return sum == K;
	if (dfs(step+1, sum)) return true;
	if (dfs(step+1, sum+a[step])) return true;
	return false;
}
int main() {
	//freopen("in.txt", "r", stdin);
	while(~scanf("%d", &N)) {
		for (int i = 0; i < N; i++) scanf("%d", &a[i]);
		scanf("%d", &K);
		memset(book, 0, sizeof(book));
		if (dfs(0, 0)) puts("Of course,I can!");
		else puts("Sorry,I can't!");
	}
	
	return 0;
}

大神剪枝代码：

#include <iostream>
#include <algorithm>
using namespace std;

bool Handle();

int src[100],n,k;
int Max[100];
int Min[100];
bool Ok;
char* result[]= {"Sorry,I can't!","Of course,I can!"};
int main(){
while(Handle()){
//empty while
}
}

void DFS(int Step, int Sum);
void InitAry();
bool Handle(){
if(!(cin>>n)){
return false;
}
for(int i= 0; i< n; ++i){
cin>>src[i];
}
cin>>k;
Ok=false;
sort(src,src+n);
InitAry();
DFS(n-1,0);
cout<<result[Ok]<<endl;
return true;
}

void DFS(int Step, int Sum){
if(Ok){//we have found one
return;
}
if(Step==-1){
if(Sum==k){
Ok= true;
}
return;
}
if(Sum+Max[Step]< k){
return;
}
if(Sum+Min[Step]> k){
return;
}
DFS(Step-1,Sum);
DFS(Step-1,Sum+src[Step]);
}

void InitAry(){
Max[0]= max(0,src[0]);
Min[0]= min(0,src[0]);
for(int i= 1; i< n; ++i){
Max[i]= max(Max[i-1],Max[i-1]+src[i]);
Min[i]= min(Min[i-1],Min[i-1]+src[i]);
}
}