hdu 1518 DFS+剪枝

Problem Description

Given a set of sticks of various lengths, is it possible to join them end-to-end to form a square?

Input

The first line of input contains N, the number of test cases. Each test case begins with an integer 4 <= M <= 20, the number of sticks. M integers follow; each gives the length of a stick - an integer between 1 and 10,000.

Output

For each case, output a line containing “yes” if is is possible to form a square; otherwise output “no”.

Sample Input

3

4 1 1 1 1

5 10 20 30 40 50

8 1 7 2 6 4 4 3 5

Sample Output

yes

no

yes

题解：

意就是好多棍子，看能不能拼成正方形。

主要注意的有几点：

所有棍子都要用到，不能剩余

输入已经保证大于4根棍子了。所以无需判断可能小于3根棍子的情况

棍长的总数首先要是4的倍数，才能进行。否则直接输出 “no”

当前面前提满足以后，再满足3 根棍子拼好，就完工了。最后一根一定能拼好。

解法就是DFS——->深度优先搜索。DFS的思路就是一个图沿着一条路走下去，当走不下去的时候就回溯到上一结点再去走没有走过的岔路。

换算成数据结构的话，就要有一个“标记”来标记每个结点是否走过。DFS具体的实现方式，常见的一种就是：循环里面嵌套递归，这也算是一个DFS的框架。而剩下的要补充的“题眼”（也就是关键的地方）是要转移的状态。

参考：http://blog.csdn.net/guodongxiaren/article/details/23126997

代码：

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

using namespace std;

bool used[21];
int stick[21];
int t,goal;

bool cmp(int a,int b)
{
return a>b;
}

bool dfs(int count,int pos,int res)
{
if(count==3)
{
return true;
}
for(int i=pos;i<t;i++)
{
if(used[i]) continue;
used[i] = true;
if(stick[i]==res)
{
if(dfs(count+1,0,goal))
return true;
}
else if(stick[i]<res)
{
if(dfs(count,i+1,res-stick[i]))
return true;
}
used[i] = false;
}
return false;
}

int main()
{
int n;
cin>>n;
while(n--)
{
cin>>t;
int sum=0;
for(int i=0;i<t;i++)
{
cin>>stick[i];
sum+=stick[i];
}
if(sum%4)
{
printf("no\n");
continue;
}
goal = sum/4;
memset(used,false,sizeof(used));
sort(stick,stick+t,cmp);
if(dfs(0,0,goal))
printf("yes\n");
else
printf("no\n");
}
return 0;
}