【DFS+剪枝】-HDU-1518-Squares

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1518

题目描述：给出若干个破木头条子。问你它们能否拼成一个正方形。

解题思路：

一道DFS题，DFS是我软肋啊。不过经过这道题，算是找到一种固定模式了，以后遇到DFS题就不会不知♂所措。

解释一下我DFS函数中的变量 len 和 dep，分别表示手里拿的木棍总长，每当达到一个边长就放下这些木棍（并标记），再从头开始找，dep表示已找到的边长数量，显然找到 4 个就能构成正方形。

交上去T了。

所以我们再考虑剪枝，问题出在我写的基本上是一个生成全排列的DFS，而这道题显然不用考虑顺序，所以每次不用从 1 开始搜，加入一个 from 变量记一下位置，下次接着搜就好了，也就是说 “排列” 和 “组合” 是不同的。这样就AC了。

我还在想，那是不是不需要 use 数组了，去掉是 WA 的，因为在你凑齐一个边长并放下手里木棍时候必须标记好用过哪些，再找的时候才是对的。

AC代码：

#include <iostream>
#include <cstring>
#include <cstdio>

using namespace std;

int arr[30],use[30];
int num,sum,found;

void dfs(int from,int len,int dep)
{
if(!found&&len<=sum/4)
{
//cout<<len<<" "<<dep<<endl;
int i;
if(len==sum/4)
{
if(dep==4)
found=1;
else
dfs(1,0,dep+1);
}
else
{
for(i=from;i<=num;i++)
{
if(!use[i])
{
use[i]=1;
dfs(i+1,len+arr[i],dep);
use[i]=0;
}
}
}
}
}

int main()
{
//freopen("1518_input.txt","r",stdin);
int T,i;
scanf("%d",&T);
while(T--)
{
scanf("%d",&num);
sum=0;found=0;
memset(arr,0,sizeof(arr));
memset(use,0,sizeof(use));
for(i=1;i<=num;i++)
{
scanf("%d",&arr[i]);
sum+=arr[i];
}
if(sum%4!=0)
{
printf("no\n");
}
else
{
dfs(1,0,1);
if(found)
printf("yes\n");
else
printf("no\n");
}
}

return 0;
}
AC截图：