POJ 1011 Sticks

DFS+减枝

所谓减枝,就是不去搜索那些不必要的。

这题的减枝方法是：将树枝从大到小排列，从最大的开始枚举。

所有长度之和sum一定要是当前枚举的倍数（才能平分）。

搜索时候，用标志位use[]记录树枝使用情况。

每次搜索，如果遇到加上其余树枝都满足不了的时候，回溯上一层。（后面的没有必要搜索了）

#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
int st[65];
int mark[65];
int s, j, n;
bool cmp(int a, int b){
return a > b;
}
bool dfs(int sum, int k, int cnt){  //sum 是正在拼的木棍已经拼出的长度， k是从哪跟碎木块开始拼，cnt是正在拼第几根木块
if (cnt == s) return true;                   //如果拼的木块数等于总长度除以所枚举的木棍长度
else if (sum == j) return dfs(0, 0, cnt + 1);   //如果本根木块拼完，拼下一根，再次从第0根木棍开始找
else{
int pre, i;
for (pre = 0, i = k; i < n; i++)                          //枚举木棍
if (mark[i] && st[i] != pre && st[i] + sum <= j){      //如果这根木棍被加过，并且它不等于前一块的长度
pre = st[i];
mark[i] = false;
if (dfs(sum + st[i], i + 1, cnt)) break;                           //从下一根木棍开始搜索
mark[i] = true;                                                             //如果向下搜索不成功回溯
if (k == 0) return false;                                                //如果回溯后k==0说明这根正在被还原的这根木棍还原失败了.也就是说剩下的木棍不足以拼出原始木棍
}
if (i == n) return false; else return true;                      //如果i < n 说明 搜索到成果了,若搜到 i == n-1 还未搜索到说明失败, 需要回退了
}
}
int main(){
int i, sum;
while(cin>>n, n){
for (i = sum = 0; i < n; i++){
cin>>st[i];
sum += st[i];
}
sort(st, st + n, cmp);
for (j = st[0]; j < sum; j++) if (sum % j == 0){                //最长的木棍的长度一定不会超过以前的木棍长度，所以从最长的长度开始枚举
s = sum / j;
memset(mark, true, sizeof(mark));
if (dfs(0, 0, 0)) break;
}
if (j == sum) cout<<sum<<endl; else cout<<j<<endl;
}
return 0;
}