【DFS】hdu 1518 Square

http://acm.hdu.edu.cn/showproblem.php?pid=1518

分析：所有木棒都要用到，刚开始完全没有领会到这个真谛

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

const int NM=25;
int a[NM],n,temp;
bool vis[NM],flag;

void DFS(int res,int ans,int k)
{
	if(flag) return;  //重要的剪枝
    if(res==temp){
        ans++;
        res=k=0;
    }
    if(ans==4){
        flag=true;return;
	}
	//k
    for(int i=k;i<n;i++){
        if(!vis[i] && res+a[i]<=temp){
            vis[i]=1;
            DFS(res+a[i],ans,i+1);
            vis[i]=0;
        }
    }
}

int main()
{
    int T,i,sum;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        sum=0;
        for(i=0;i<n;i++){
            scanf("%d",&a[i]);
            sum+=a[i];
        }
        if(sum%4!=0) {
            printf("no\n");continue;
        }
        temp=sum/4;
        for(i=0;i<n;i++){
            if(a[i]>temp) {
                printf("no\n");continue;
            }
        }
        
        flag=false;
        memset(vis,0,sizeof(vis));
        DFS(0,0,0);
        if(flag) printf("yes\n");
        else printf("no\n");
    }
    return 0;
}