为什么这道题用深搜?
我们可以把所有的小木棒当做图的一个数据节点,因为每一个木棒都可以和任意木棒搭配(是否符合题意另当别论),所以这个图是一个完全无向图。
我们目的是要在这个完全无向图里,寻找四条符合条件、不会相交的路径。使用广搜会有相当多不必要的遍历,用深搜最符合。
上代码:
/*
Square
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
*/
#include<iostream>
#include<algorithm>
int n, m, des;
int a[21];
bool used[21];
using namespace std;
bool cmp(int a, int b) {
return a > b;
}
//cur 当前走到第几根
//done 已经完成的边数
//res 该边剩余未完成
bool dfs(int cur,int done,int res) {
//已经完成三边,最后一边一定可以完成
if (done == 3)
return true;
for (int i = cur; i < m; i++) {
//这一根已经被用过
if (used[i])
continue;
used[i] = true;
if (a[i] == res) {
//保证每一根都用到
if (dfs(0, done + 1, des))
return true;
}
else if(a[i] < res) {
//为什么这里不像上面从0开始呢
//深搜的意义也在于此:
//当前路径已走到尽头就向上一级重新深搜,
//上面从0开始是因为那条路径已经满足情况了,没有必要继续补充那条边
if (dfs(i + 1, done, res-a[i]))
return true;
}
//这一根不能独自完成,与比它小的且未搭配的也不能完成。回溯,暂时不用他。
//等他的搭档被释放在使用
used[i] = false;
}
//始终找不到合适的路径
return false;
}
int main() {
cin >> n;
while (n--) {
cin >> m;
int sum = 0;
memset(a, 0, sizeof(a));
memset(used, false, sizeof(used));
for (int i = 0; i < m; i++) {
cin >> a[i];
sum += a[i];
}
//知道木棍总长,我们要围成一个矩形,故边长已知
des = sum / 4;
if (des * 4 != sum) {
cout << "no" << endl;
continue;
}
//把木棍们从大到小排序好
sort(a, a + m, cmp);
if (dfs(0,0,des))
cout << "yes" << endl;
else
cout << "no" << endl;
}
return 0;
}