【POJ】Square(DFS + 剪枝)

记录一个菜逼的成长。。

题目大意：
给你一些棍子的长度，问这些棍子能否组成正方形。
剪枝：
1.既然是正方形，长度总和肯定能被4整除。
2.最长的棍子的长度一定小于等于正方形边长。
3.满足上述两个条件，只要判断三边是否符合就可。

看到网上一些题解上有对边长的排序，其实不排序也能过。

#include <cstdio>
#include <iostream>
#include <cstring>
#include <string>
#include <algorithm>
#include <cstdlib>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <list>
#include <deque>
#include <cctype>
#include <bitset>
#include <cmath>
using namespace std;
#define ALL(v) (v).begin(),(v).end()
#define cl(a) memset(a,0,sizeof(a))
#define bp __builtin_popcount
#define pb push_back
#define fin freopen("D://in.txt","r",stdin)
#define fout freopen("D://out.txt","w",stdout)
#define lson t<<1,l,mid
#define rson t<<1|1,mid+1,r
#define seglen (node[t].r-node[t].l+1)
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int,int> PII;
typedef pair<LL,LL> PLL;
typedef vector<PII> VPII;
const int INF = 0x3f3f3f3f;
const int MOD = 1e9 + 7;
template <typename T>
inline void read(T &x){
    T ans=0;
    char last=' ',ch=getchar();
    while(ch<'0' || ch>'9')last=ch,ch=getchar();
    while(ch>='0' && ch<='9')ans=ans*10+ch-'0',ch=getchar();
    if(last=='-')ans=-ans;
    x = ans;
}
inline bool DBread(double &num)
{
    char in;double Dec=0.1;
    bool IsN=false,IsD=false;
    in=getchar();
    if(in==EOF) return false;
    while(in!='-'&&in!='.'&&(in<'0'||in>'9'))
        in=getchar();
    if(in=='-'){IsN=true;num=0;}
    else if(in=='.'){IsD=true;num=0;}
    else num=in-'0';
    if(!IsD){
        while(in=getchar(),in>='0'&&in<='9'){
            num*=10;num+=in-'0';}
    }
    if(in!='.'){
        if(IsN) num=-num;
            return true;
    }else{
        while(in=getchar(),in>='0'&&in<='9'){
                num+=Dec*(in-'0');Dec*=0.1;
        }
    }
    if(IsN) num=-num;
    return true;
}
template <typename T>
inline void write(T a) {
    if(a < 0) { putchar('-'); a = -a; }
    if(a >= 10) write(a / 10);
    putchar(a % 10 + '0');
}
/******************head***********************/
const int maxn = 20 + 10;
int a[maxn],vis[maxn],edge,n;
bool dfs(int x,int sum,int cnt)//下标，边长，计数
{
    if(cnt == 3)return true;//满足三边
    for( int i = x; i <= n; i++ ){
        if(!vis[i]){
            vis[i] = 1;
            if(sum + a[i] < edge){
                if(dfs(i,sum+a[i],cnt))return true;
            }
            else if(sum + a[i] == edge){
                //**若等于边长，则需要从头开始搜索，才不会漏掉**
                if(dfs(1,0,cnt+1))return true;
            }
            vis[i] = 0;
        }
    }
    return false;
}
int main()
{
    int T;read(T);
    while(T--){
        cl(vis);
        int sum = 0,mx = 0;read(n);
        for( int i = 1; i <= n; i++ ){
            read(a[i]);
            mx = max(mx,a[i]);
            sum += a[i];
        }
        edge = sum / 4;
        if(sum % 4 || mx > edge){
            printf("no\n");
            continue;
        }
        printf("%s\n",dfs(1,0,0) ? "yes":"no");
    }
    return 0;
}