木棒

题目描述

输入样例：

9
5 2 1 5 2 1 5 2 1
4
1 2 3 4
0

输出样例：

6
5

【思路】

优化

【AC代码】

#include <iostream>
#include <algorithm>
#include <cstring>using namespace std;const int N = 70;int w[N], sum, length, n;
bool st[N];bool dfs(int u, int cur, int start) {	//当前拼接木棒个数，当前木棒长度 ，从第几个已有木棒进行枚举搜索 if(u * length == sum)	return true;// 如果当前木棒的数量 * 长度 = sum，则找到了答案，直接返回 trueif(cur == length) return dfs(u + 1, 0, 0);// 如果当前木棒的长度等于了 length，则新开一根木棒// 否则枚举从 start 开始枚举将哪根木棍拼到当前木棒中for(int i = start; i < n; i ++) {//若已经搜索过 if(st[i] || cur + w[i] > length)	continue;//若没有搜索过st[i] = true;// 继续搜索当前木棒的下一根木棍拼哪根if(dfs(u, cur + w[i], i + 1))	return true;//恢复现场st[i] = false;// 执行到这 dfs(u, s + w[i], i + 1) 为 false: 说明当前木棍搜索失败了//继续进行剪枝优化//如果第一根或者最后一根搜索失败，那么一定失败if(cur == 0 || cur + w[i] == length)	return false;int j = i;
// 		如果当前木棍没有搜到方案，则跳过所有长度相等的木棍while(j < n && w[j] == w[i])	j ++;i = j - 1; }	//若是上边没有匹配成功，则返回失败return false;
}int main() {while(cin >> n, n) {memset(st, 0, sizeof st);sum = 0, length = 1; // 每根木棒的长度从 1 开始枚举，从小到大枚举，题目要求分组最多，那么第一个满足条件的就是答案for(int i = 0; i < n; i ++) {cin >> w[i];sum += w[i];}//从大到小枚举，减少搜索分支 sort(w, w + n, greater<int>());while(true) {// 循环一定会退出，最坏情况下所有木棍拼一根木棒if(sum % length == 0 && dfs(0, 0, 0)) {cout << length << endl;break;}length++;// 当前 length 无法得到答案，则长度 + 1}}return 0;
}