square（正方形）

poj 2362

题目大意：给出一些数，问你能不能把这些数，分成4堆，没堆的和相等

解决：dfs+回溯，此题用的剪枝稍微少一些就不行

#include <iostream> 

#include <cstring>

#include <algorithm>

#include <cstdio>

#include <functional>

using namespace std;

int num[30],ave,n;



bool found;

bool vis[30];



bool check()

{

    for(int i=0;i<n;i++)if(!vis[i])return false;

    return true;

}



void dfs(int p,int cur_sum,int heap)

{

    if(found)return;

    if(heap==4 && check()){  found=true;  return; }

    for(int i=p;i<n;i++)

    {

        if(!vis[i] && cur_sum+num[i]<=ave )

        {

             vis[i]=1;

             if(cur_sum+num[i]==ave){if(heap+1<=4) dfs(0,0,heap+1);}

             else dfs(i+1,cur_sum+num[i],heap);

             vis[i]=0;

        }

    }

}

int main()

{

    int T;

    scanf("%d",&T);

    while(T--)

    {

        ave=0;

        found=false;

        memset(vis,0,sizeof(vis));

        scanf("%d",&n);

        for(int i=0;i<n;i++){scanf("%d",num+i); ave+=num[i];} 

       //必须被4整除

        if(ave%4)puts("no");

        else

        {

            

            ave/=4;

            //按照从大到小排序

            sort(num,num+n,greater<int>()); 

           //必须所有的都小于平均数

            if(num[0]>ave)puts("no");

            else 

            { 

                dfs(0,0,0);

                if( found )puts("yes");

                else puts("no");

            }

         } 

    }

    system("pause");

    return 0;

}