ZOJ 1909 Square(DFS+剪枝)

开始对边排一下序,

搜索的时候完整的拼成了一条边的话从0开始搜索下一条完整的边.

否则因为前面的木棍要么已经被选择了，要么是长度比当前完整边剩下的要长,根本没有必要再从0开始

所以可以从当前的小木棍的下一个条小木棍开始搜索完成当前这条边,.不加这个剪枝怎么都过不了,20太大了.

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <functional>
#include <memory.h>
using namespace std;
const int maxn=25;
int sticks[maxn],tarLen,n,sm;
bool vis[maxn];
bool dfs(int allSum,int rest,int index){
	if(allSum==sm)return true;
	if(rest>tarLen)return false;
	for (int i=index;i<n;++i){
		if(!vis[i]){
			vis[i]=1;
			if(rest-sticks[i]==0){//形成了一条完整的边,重新开始下一条完整的边
				if(dfs(allSum+sticks[i],tarLen,0))return true;
			}else if(sticks[i]<=rest){//下一条木棍直接从i开始搜
				if(dfs(allSum+sticks[i],rest-sticks[i],i))return true;
			}
			vis[i]=0;
		}
	}
	return false;	
}


int main(){
	int t;
	scanf("%d",&t);
	while (t--){
		scanf("%d",&n);
		int sum=0;
		for (int i=0;i<n;++i){
			scanf("%d",&sticks[i]);
			sum+=sticks[i];
		}
		memset(vis,0,sizeof(vis));
		sort(sticks,sticks+n,greater<int>());
		tarLen=sum/4;
		if(sum%4!=0||sticks[0]>tarLen){//如果边长长度不是整数或者小棍子里最长的比边长还长,没有必要搜索
			printf("no\n");
		}else{
			sm=sum;
			if(dfs(0,tarLen,0)){
				printf("yes\n");
			}else{
				printf("no\n");
			}
		}

	}
	return 0;
}