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组边即可。主要是代码实现比较难。而且很容易T,是一道策略+dfs+剪枝的题目。

策略就是用点贪心策略,把树枝都排一下序,若选一个数枝超了正方形边长那么就选比它小的,所以搜索的范围就缩小了。时间缩短,而且也可以借此用两个剪枝:

读入的最长树枝若超了正方形边长直接no。如果以当前最长的没有被选取的树枝开始拼一条新边却失败了,则不可能成功(需要动脑)。

这个策略中还有一个贪心思想,如果一根长度和剩余长度一样的木棒,就一定可以把他们配起来,如果配起来之后,最终不能形成正方形,就不必继续找比当前匹配失败的更小的木棒继续配, 可以断言,配对必定失败,因为这是一种基于这样的贪心思想:如果一根长度和剩余长度一样的木棒,就一定可以把他们配起来,因为留下了一些短小的木棒更灵活更利于配其他边,这是灵活度利用的贪心,不妨用极限来理解,假设每根木棒无穷短,那么一定可以配成正方形,因为太短了,灵活性太大。

接下来给几个剪枝:


  剪枝
  1:树枝的长度和为4的倍数
  2:树枝的长度和除以4要大于等于最长的树枝的长度
  3:将树枝从大到小排序
  5:如果当前树枝不能被选取,则与当前树枝等长的树枝也不能被选取
  6:如果以当前最长的没有被选取的树枝开始拼一条新边却失败了,则不可能成功。       //好剪枝较难想到。
  7:如果一根长度和剩余长度一样的木棒,把他们配起来,如果配起来之后,最终不能形成正方形,那么不可能成功。 //好剪枝较难想到。

当然这些剪枝不需全加上去,加一部分就可以了。

上c++代码:

#include
#include
#include
#include
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)
//条件可以写成 if(l*4!=sum) 时间会从73ms升到二百多ms,但是去掉l*4!=sum这个条件 以后就会wa。因为只可能是l*4

本题的剪枝也是很多的。

上大仙的剪枝代码(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 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
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;ib;
}
int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        scanf("%d",&n);
        sum=0;
        Max=0;
        for(int i=0;iMax) 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;
}




你可能感兴趣的:(DFS)