POJ1011洛谷过了,但POJ上WA了。
https://www.luogu.com.cn/paste/dk6r5t9o
你看一下这个行不行:
import java.util.ArrayList;
import java.util.Collections;
import java.util.Scanner;
/**
* created by 吴家俊 on 2019/9/17.
*/
public class Main{
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
int count = scanner.nextInt();
ArrayList<Integer> list = new ArrayList();
int sum;
while (count != 0) {
sum = 0;
for (int i = 0; i < count; i++) {
int k = scanner.nextInt();
list.add(k);
sum += k;
}
Bar bar = new Bar(list, sum);
bar.inputResult();
count = scanner.nextInt();
//清空list
list.clear();
}
}
}
class Bar {
private ArrayList<Integer> list;//存储被切后各个棒的长度
private int[] visit; //访问数组标记某个数是否被使用过 0未使用 1使用
private int len; //预测最初每个棒的长度
private int n; //最初棒的个数
private int sum; //棒的总长度
private boolean flag; //标记是否得出结果
public Bar(ArrayList<Integer> list, int sum) {
this.list = list;
this.sum = sum;
Collections.sort(list);
Collections.reverse(list); //按棒长降序排序
}
private void init() {
visit = new int[list.size()];
flag = false;
}
public void inputResult() {
for (int i = list.get(0); i <= sum; i++) { //1、缩小len的取值范围
if (i == sum) {
System.out.println(i);
} else if (sum % i == 0) { //2、再次缩小len的取值范围
n = sum / i;
len = i;
init();
dfs(0, len, 0);
if (flag) {
break;
}
}
}
}
/**
* @param index 遍历的起始下标
* @param l 当前剩余长度 为0时 表示凑成了一个原先的棒
* @param now 当前已经凑成的棒的根数 为n时 表示得到解
*/
private void dfs(int index, int l, int now) {
if (flag) return;
if (now >= n) {
flag = true;
System.out.println(len);
return;
}
for (int i = index; i < visit.length; i++) {
if (visit[i] == 0 && l - list.get(i) >= 0) {
visit[i] = 1; //标记已访问
if (l - list.get(i) == 0) { //凑成了一根棒
dfs(0, len, now + 1); //开始寻找下一个棒的起点
} else {
dfs(index + 1, l - list.get(i), now);
}
visit[i] = 0; //还原
if (flag) return;
if (l == len) return; //3、之前的决策有问题,回退到上一根棒的拼凑过程
while (i + 1 < visit.length && list.get(i + 1) == list.get(i)) {//4、该点失败后,后面相同的点不用再尝试
i++;
}
}
}
}
}