hdu 1518 又一道搜索经典

刚刚做过hdu 1455那道搜索题，在来做这道，那就像切菜一样。。。，orz，让我这种若菜也体验了一把切题的乐趣。。。。

好吧，还是说一下大意吧。。。这次是为正方形了。。。就是所有的木棒从头到尾相连，判断能否组成一个正方形。。。搜索过程没有像1455那题有那么多的剪枝

。。。orz，我是直接暴力的。。。还是上代码吧。。。

 

#include<iostream>
#include<algorithm>
using namespace std;

struct stick{
    int length;   //长度
    int mark;    //标记是够被使用过
};
stick sticks[21];
int n,len;

int cmp(const stick &s1,const stick &s2){
    return s1.length>s2.length;
}
//len为所围成的正方形的边长，l为当前木棒的长度
int dfs(int len,int l,int count,int pos){
    if(count==4)return 1;
    for(int i=pos;i<n;i++){
        if(sticks[i].mark)continue;
        //找到了相等的
        if(len==(l+sticks[i].length)){
            sticks[i].mark=1;
            if(dfs(len,0,count+1,0))return 1;
            sticks[i].mark=0;
        }else if(len>(l+sticks[i].length)){
            sticks[i].mark=1;
            l+=sticks[i].length;
            if(dfs(len,l,count,i+1))return 1;
            l-=sticks[i].length;  //不符合的话要恢复原长
            sticks[i].mark=0;
        }
    }
    return 0;
}

int main(){
    int t;
    scanf("%d",&t);
    while(t--){
        scanf("%d",&n);
        len=0;
        for(int i=0;i<n;i++){
            scanf("%d",&sticks[i].length);
            len+=sticks[i].length;
            sticks[i].mark=0;
        }
        //不是4的倍数的以及木棒的根数少于4的都不能
        if(len%4||n<4){
            printf("no\n");
            continue;
        }
        len/=4;
        sort(sticks,sticks+n,cmp);
        if(dfs(len,0,0,0)){
            printf("yes\n");
        }else 
            printf("no\n");
    }
    return 0;
}