acm 2362 square
http://acm.pku.edu.cn/JudgeOnline/problem?id=2362
给定一些有特定长度的小棍，问能否拼成正方形。很经典的深度优先搜索+剪枝的题目。
1. 首先不能被4 整除的不行
排序后最长的木棍大于总长度/4的也不行，也就是大于唯一可能的正方形边长的。
2. 然后就是深度优先搜索，
举例说如果，长度总和，除以 4 就是正方形边长，我们设定为len
我们要找出是否存在一个分组，将小木棍分成四组，每组长度和为len
3. 两个典型的测试用例
1 3 4 5 4 7 8 可以 1 7 3 5 4 4 8 注意不能 1 3 4 凑成一组
5 4 7 8 8 不可以
4. 基本的DFS搜索思路
对于每个木棍，可以选或者不选两种状态，顺序扫描所有的木棍，然后找出所有可能的组合，使得长度和可能为len
(对于给的例子即为8)，
对于一个给定的长度为8的木棍组，继续递归寻找另一组，直到能找到所有组则成功，但是如果后续失败了，
还能返回上面，继续尝试
如 找到 1 3 4 然后 找到 8 然后不成功
还可以退回来 找到 1 7
5. 剪枝优化
5.1.其实我们只要找3个分组，对于总和为32的来说，找3个长度为8 的分组成功的话就成功了
5.2. 每次选组的时候，首元素确定。为什么要确定首元素，剪枝，（和排序无关）
因为本质上我们要找的4个分组是不分顺序的！！
abc def ghi jkl
和 def jkl ghi abc是一个解 找到就可以
所以 对于 a 开头的组
如果我们试图找 (a,..) (...)(...) (...)失败了 那么没有必要找 (!a,....)(...)(...)(...)了
假如能找到 后面的成立 那么 a 必然在后面某一组 ，调到前面来也应该成立 矛盾了
这样每次找某个组的时候剪枝50%
5.3. 还有什么地方可以剪枝呢（这里也和排序无关）
当到达某个元素选定它的时候正好长度达到要求，如上面的例子 总和32，这个选的分组包含该元素正好长度为8.
1 3 4 4 5 7 8
扫描到4 发现长度和为8了 然后 找 （...) (...) (...)失败了
这时候只要退出到上一层继续就好了
不需要考虑 该层不选4的情况
当然对于 从大到小排列 很清楚 后面的更大 更不可能
6 2 1 1 1 1 1 1 1 1 1 1
对于这种情况，假如 扫描到2 发现正好为8，然后查找后续3组（．．．） （．．．）(...)失败
同样可以剪枝，不需要考虑 6 ！2 。。。。。。。。。。。。 这种情况
因为加入后面的需要考虑 那么 令 x = 2 2前面的集合为 U
sum( U + x) = len
最后假设分组成功 ，肯定 存在某个分组 sum(U + Z) = len sum(V+x) = len
那么完全可以互换 Z组和 x，这于前面 U+x 然后向后寻找其他组失败是矛盾的。
5.4.为什么要从大到小排列
POJ的测试数据证明了从大到小排序，快速的多，意味着更少的搜索。
首先从大到小，是先找数目少的集合，能较快的确定某个长度为len 的组，
而且确定的过程中经过许多单分支（仅 false)
8 7 5 4 4 3 1
如 找到 7 1
true false false false false true
而考虑 1 3 4 4 5 7 8
会先找到 1 3 4 后续分组不成功 退回来找 到 1 7
这个过程大量的搜索因为绝大多数分组都是2分的（选中或者不选），两个向下的分支。
下面注意是考虑从小到大排序， 0 右子树也会被剪枝掉，
0 7
1 0 7
4 1 4 0 7
8 4 5 1 8 4 4 0 7
4 9 5 7
12 5 8
5.5 找到后立即退出，也算剪枝。
方法.1 抛出异常
2. 适当的 return 语句 if (dfs(next)) return, 这样成功后立即向上逐层直接退出而不进入后续分支
choosen()
if (dfs(next) return true
not choosen()
if dfs(next) return true
return false
5. 6 一个小的剪枝
当两个元素相同，排序后紧挨着，前面的元素如果为被选中，后面的也不需要选。
否则重复了。
4 ！4
！4 4 是相同的情形。
//代码主要参考了http://www.cppblog.com/jhpjhyx/archive/2009/05/14/82981.html 表示感谢：）
#include <iostream>
#include <algorithm> using namespace std; int test_num; int sl[20]; bool choosen[20]; int stick_num; bool find_square; int sum; int len; struct cmper{ inline bool operator ()(const int &a, const int &b)const{ return a > b; } }; //确保了最大的要小于＝ len,也正确 但递归调用较多，考虑用下面的循环减少递归 //void find_group(int set_num, int depth) { // if (depth == stick_num) // return; // if (set_num == 0) { // find_square = true; // throw true; // } // // if (choosen[depth] == true) // find_group(set_num, depth + 1); // else { // if (depth && !sl[depth - 1] && sl[depth] == sl[depth -1]) { // find_group(set_num, depth + 1); // return; // } // int sum_now = sl[depth] + sum; // if (sum_now < len) { // choosen[depth] = true; // sum = sum_now; // find_group(set_num, depth + 1); // sum = sum_now - sl[depth]; // choosen[depth] = false; // if (sum != 0) //剪枝，最大的只能被选中，不需要考虑不被选中的情况，剪枝50% // find_group(set_num, depth + 1); // } // else if (sum_now == len) { // choosen[depth] = true; // sum = 0; // find_group(set_num - 1, 0); // choosen[depth] = false; // // } // else { // find_group(set_num, depth + 1); //sum_now > len 只能不选当前的 // } // // } //} bool find_group(int set_num, int depth , int sum = 0) { if (set_num == 1) //选好3组了，返回成功 return true; for (int i = depth; i < stick_num; i++) { if (choosen[i] == true) //前面的选好的组已经用到该元素，略过继续查找。 continue; if (i && !sl[i - 1] && sl[i] == sl[i -1]) //剪枝,作用不大 continue; int sum_now = sl[i] + sum; if (sum_now < len) { choosen[i] = true; //选中当前元素，继续后续查找当前组组元素 if (find_group(set_num, i + 1, sum_now)) return true; choosen[i] = false; //不选中当期元素的情形(注意向上返回时都已置回false),继续后续查找当前组元素 } else if (sum_now == len) { choosen[i] = true; if (find_group(set_num - 1, 0, 0)) //找到1组，查找后续分组，成功的话返回true return true; choosen[i] = false; } if (sum_now == len || sum == 0) //剪枝，首元素要被选中，当包含首元素的组查找其它分组失败即为失败5.2 break; //当选中当前元素，能构成一个组的时候，如果查找后续分组失败即为失败5.3 } return false; //当前选择的情形，已经扫描到末尾仍为成功失向上层返回标识为失败 } bool group_stick() { find_square = false; for (int i = 0; i < 20; i++) choosen[i] = false; //try{ return find_group(4, 0); //} catch(bool nothing){ } //return find_square; } int main(int argc, char *argv[]) { cin >> test_num; int sum; for (int i = 0; i < test_num; i++) { cin >> stick_num; sum = 0; for (int j = 0; j < stick_num; j++) { cin >> sl[j]; sum += sl[j]; //choosen[i] = false; } if ((sum % 4) != 0) cout << "no" << endl; else { sort(sl, sl + stick_num, cmper()); len = sum / 4; if (sl[0] > len) cout << "no" << endl; else { if (group_stick()) cout << "yes" << endl; else cout << "no" << endl; } } } return 0; }