HDU 1518 Square 深搜

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

题意：

　　给出M根棍子，要将所有棍子用上，拼成一个正方形。

 

坑爹：

　　用深搜时递归容易超时。

 

解法：

　　剪枝：用一个pos来记录当前搜索到的位置，当下次继续加一个棍子时从pos位置开始，因为前面的可以加的都已经标记过了，而每递归一次都要从头

开始搜索，浪费了很多时间。

 

#include<iostream>
using namespace std;

const int maxn = 20 + 10;
int num[maxn];
bool used[maxn];        
int M;
int sum;
int flag;

int DFS(int d,int cnt,int pos)
{
    if(d == sum)
    {
        d=0;
        pos=0;
        cnt++;
    }
    if(cnt == 4)
    {
        flag = 1;
        return 1;
    }
    int i;
    for(i=pos; i<M; i++)
    {
        if(!used[i])
        {
            used[i] = 1;
            if(num[i]+d <= sum)
            {
                if(DFS(num[i]+d,cnt,i+1) == 1)
                {
                    return 1;
                }
            }
            used[i] = 0;
        }
    }
    return 0;
}

int main()
{
    int N;
    cin>>N;
    while(N--)
    {
        memset(used,0,sizeof(used));
        int i;
        sum = 0;
        cin>>M;
        for(i=0; i<M; i++)
        {
            cin>>num[i];
            sum += num[i];
        }
        if(sum%4 != 0)
        {
            cout<<"no"<<endl;
            continue;
        }
        sum/=4 ;
        flag=0;
        DFS(0,0,0);
        if(flag)
        {
            cout<<"yes"<<endl;
        }
        else
        {
            cout<<"no"<<endl;
        }
    }
    return 0;
}