POJ2362 Square

题目描述：给定一系列枝条，判断他们是否可以收尾相连接成一个正方形。

样例输入：

第一行是样例个数N，以下各n行是的第一个数M是这个样例的枝条数，然后后面跟着各个枝条的长度。

限制条件：

4<=M<=20

每个枝条的长度between 1 and 10,000。

样例i/o：

Sample Input

3
4 1 1 1 1
5 10 20 30 40 50
8 1 7 2 6 4 4 3 5

Sample Output

yes
no
yes

思路：dfs，也是我目前做过的最难的一道dfs，思路都显然知道：深搜找到4组边即可。主要是代码实现比较难。因为没接触过四组和标记的消除问题。

上c++代码：

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
using namespace std;
int map[21]={0},stick[21],l,sum,N;
bool dfs(int n,int s,int len)
{
	if (n==3) return true;                    //不需要找到四组，三组已找到，四组必然。
	for(int i=s;i>=0;i--)
	{
		if(!map[i])
		{
			map[i]=1;
			if(len+stick[i]<l&&dfs(n,i-1,len+stick[i]))
				return true;
			if(len+stick[i]==l&&dfs(n+1,N-1,0))
				return true;
			else map[i]=0;                          //标记的实现和消除是难点。
		}

	}
	return false;
}

int main()
{
	int _;
	scanf("%d",&_);
	while(_--)
	{
		sum=0;
		memset(map,0,sizeof(map));
		memset(stick,0,sizeof(stick));
		scanf("%d",&N);
		for(int i=0;i<N;i++) {
			scanf("%d",&stick[i]);
			sum+=stick[i];

		}
		l=sum/4;
		sort(stick,stick+N);
		if(l*4!=sum||stick[N-1]>l)

<span style="white-space:pre">						</span>//条件可以写成 if(l*4!=sum) 时间会从73ms升到二百多ms，但是去掉l*4！=sum这个<span style="white-space:pre">							</span>条件 以后就会wa。因为只可能是l*4<sum，所以其中有可能有一个或者几个树棍没计在内。
		{
			printf("no\n");
		}

		else if(dfs(0,N-1,0)) printf("yes\n");
		else printf("no\n");

	}
	return 0;
}

本题的剪枝也是很多的。

上大仙的剪枝代码（0ms运行）：

/*
 * 剪枝
 * 1:树枝的长度和为4的倍数
 * 2:树枝的长度和除以4要大于等于最长的树枝的长度
 * 3:将树枝从大到小排序
 * 5:如果当前树枝不能被选取，则与当前树枝等长的树枝也不能被选取
 * 6:如果以当前最长的没有被选取的树枝开始拼一条新边却失败了，则不可能成功。       //好剪枝。
 */
/*************************************************************************
 > File Name: 2362.cpp
 > Author: UnknownCUnknown
 > Mail: [email protected]
 > Created Time: 二 12/16 00:08:45 2014
 ************************************************************************/

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cctype>
#include <vector>
#include <map>
#include <set>
#include <stack>
#include <list>
#include <string>
#include <cstdlib>
#include <queue>
#include <cmath>
#include <climits>
using namespace std;
bool vis[25];
int len[25];
int n;
int sum,side;
int Max=0;
bool flag1;
void dfs(int le,int now,int num){
    if(le==side) {
        ++num;
        now=0;
        le=0;
    }
    if(num==3){
        flag1=true;
    }
    if(le>side){
        return;
    }
    if(flag1) return;
    for(int i=now;i<n;++i){
      //  if(flag1) return;
        if(i&&!vis[i-1]&&len[i]==len[i-1]) continue;//剪枝5
        if(!vis[i]){
            vis[i]=true;
            dfs(le+len[i],i+1,num);
      //      if(flag1) return;
            vis[i]=false;
            if(le==0) return;
        }
    }
}
bool cmp(int a,int b){
    return a>b;
}
int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        scanf("%d",&n);
        sum=0;
        Max=0;
        for(int i=0;i<n;++i){
            scanf("%d",len+i);
            if(len[i]>Max) Max=len[i];
            sum+=len[i];
        }
        if(sum%4!=0){//剪枝1
            printf("no\n");
            continue;
        }
        side=sum/4;
        if(Max>side) {//剪枝2
            printf("no\n");
            continue;
        }
        sort(len,len+n,cmp);//剪枝3
        flag1=false;
        memset(vis,false,sizeof(vis));
        dfs(0, 0,0);
        if(!flag1) printf("no\n");
        else printf("yes\n");
    }
    return 0;
}