给出一个数组A包含n个元素,表示n本书以及各自的页数。现在有个k个人复印书籍,每个人只能复印连续一段编号的书,比如A[1],A[2]由第一个人复印,但是不能A[1],A[3]由第一个人复印,求最少需要的时间复印所有书。
神奇的二分法:
public int copyBooks(int[] pages, int k) {
if (pages == null || pages.length == 0) {
return 0;
}
int total = 0;
int max = 0;
for (int i = 0; i < pages.length; i++) {
total += pages[i];
max = Math.max(max, pages[i]);
}
int start = max;
int end = total;
while (start + 1 < end) {
int mid = start + (end - start) / 2;
if (copier(pages, mid) > k) {
start = mid;
} else {
end = mid;
}
}
if (copier(pages, start) <= k) {
return start;
}
return end;
}
private int copier(int[] pages, int limit) {
if (pages == null || pages.length == 0) {
return 0;
}
int work = pages[0];
int copier = 1;
for (int i = 1; i < pages.length; i++) {
if (work + pages[i] > limit) {
work = 0;
copier++;
}
work += pages[i];
}
return copier;
}
对答案进行二分,可以参照wood cut一题,还是二分法的主体框架,但是在判决条件上做文章,需要好好理解理解。