POJ 2362 dfs 剪枝

题目大意：给你若干木棍的长度， 是否能组成正方形。

n必须大于等于4， sum必须能被4整除。

#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
#include <string>
#include <map>
#include <iomanip>
#include <cstdio>
#include <iostream>
using namespace std;
const int N = 30;
const int inf = 0x3f3f3f3f;
int side, n;
int num[N];
bool vis[N];
bool dfs(int m, int len, int cur)
{
    if(m==3)//回溯的条件
       return true;

    for(int i = cur; i<n; i++)
    {
        if(vis[i])
            continue;
        vis[i] = true;
        if(len+num[i]<side)
        {
            if(dfs(m, len+num[i], i))
                return true;
        }
        if(len + num[i] == side)
        {
            if(dfs(m+1, 0, 0))
                return true;
        }
        vis[i] = false;
    }
    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", &num[i]);
            sum+=num[i];
        }
        if(sum%4==0&&n>=4)
            side = sum/4;
        else
        {
            printf("no\n");
            continue;
        }
        memset(vis,false, sizeof(vis));
        if(dfs(0,0,0))
            printf("yes\n");
        else
            printf("no\n");
    }
    return 0;
}