【深度搜索+剪枝】POJ2362-Square

从昨天晚上写到现在，一直在TLE，现在终于剪枝完成了_(:зゝ∠)_

【思路】

深搜：用这类型组合题目最基本的深搜，变量side记录当成已经组成了几条变，sl表示当前在组合的边已经有的长度。如果当前stick的长度与已有长度的和恰巧等于边长，则side+1，将sl清零；否则若小于边长，则sl+当前长度继续搜索，直到组成所有边为止。

剪枝：（1）如果木棒数目没有到达四根，则为no

　　   （2）比较容易想到的一点，如果当前木棒总长不是4的整数倍，则为no

　　　（3）由于木棒不能这段，如果最长的木棒大于边长，则为no

　　　（4）由于越短的木棒灵活性越高，进行快排之后由大至小进行搜索。

　　　（5）为了避免重复搜索，我们默认组成同一边时，下一根取的木棒长度必定不大于当前根的木棒。故设置变量frm，即当前可取木棒长度的范围。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN=20+10;
int stick[MAXN];
int n,sum,m; 
bool visited[MAXN];

bool dfs(int frm,int side,int sl)
{
    if (3==side) return(true);
    for (int i=frm;i>=0;i--)
        if (!visited[i])
        {
            visited[i]=true;
            if (stick[i]+sl<sum)
            {
                if (dfs(i-1,side,sl+stick[i])) return true;
            }
            else
                if (stick[i]+sl==sum)
                {
                    if (dfs(m-1,side+1,0)) return true;
                }
            visited[i]=false;
        }
    return false;
}

int main()
{
    scanf("%d",&n);
    for (int kase=0;kase<n;kase++)
    {
        memset(visited,false,sizeof(visited));
        scanf("%d",&m);
        sum=0;
        int max=-1;
        for (int i=0;i<m;i++) 
        {
            scanf("%d",&stick[i]);
            sum+=stick[i];
            if (stick[i]>max) max=stick[i];
        }
        if (sum%4!=0||m<4||max>sum/4) cout<<"no"<<endl;
        else
        {
            sort(stick,stick+m);
            sum=sum/4;
            if (dfs(m-1,0,0)) cout<<"yes"<<endl;
            else cout<<"no"<<endl;
        }
    }
    return 0;
}