POJ 2362 Square

一道简单的搜索剪枝，居然写了一个小时，首先判断棍子长的和是否能被四整除以及最长

的棍子是不是大于边长（和的四分之一）。将棍长按照从长到短排序，记录搜索的边，构成

三条边就是构成了一个正方形。

/*Accepted    164K    235MS    C++    1396B    2012-07-27 14:30:28*/

#include<cstdio>

#include<cstring>

#include<cstdlib>

#include<algorithm>

using namespace std;



const int MAXN = 1 << 7;

int m, a[MAXN], sum, side, flagcnt;

bool flag, used[MAXN];



bool cmp( const int i, const int j)

{

    return i > j;

}



void init()

{

    scanf( "%d", &m);

    sum = 0;

    flag = false;

    memset( used, false, sizeof used);

    for( int i = 1; i <= m; i ++)

    {

        scanf( "%d", &a[i]);

        sum += a[i];

    }

    side = sum / 4;

}



void dfs( int st, int now, int flagcnt)

{

    if(flagcnt == 3){

        flag = true;

        return;

    }

    if(now == side) dfs( 1, 0, flagcnt + 1);

    if(flag) return;

    for( int i = st; i <= m; i ++)

    {

        if( !used[i] && now + a[i] <= side)

        {

            used[i] = true;

            dfs( i + 1, now + a[i], flagcnt);

            if(flag) return;

            used[i] = false;

        }

    }

}





bool judge()

{

    if(a[1] > side || sum % 4 != 0)

        return false;

    return true;

}



int main()

{

    int T;

    scanf( "%d", &T);

    while( T --)

    {

        init();

        if( !judge()) {

            printf( "no\n");

            continue;

        }

        sort( a + 1, a + m + 1, cmp);

        dfs(1, 0, 0);

        if( flag)

            printf( "yes\n");

        else

            printf( "no\n");

    }

    return 0;

}