有一些原木,现在想把这些木头切割成一些长度相同的小段木头,需要得到的小段的数目至少为 k。当然,我们希望得到的小段越长越好,你需要计算能够得到的小段木头的最大长度。
注意事项:
木头长度的单位是厘米。原木的长度都是正整数,我们要求切割得到的小段木头的长度也要求是整数。无法切出要求至少 k 段的,则返回 0 即可。
样例
有3根木头[232, 124, 456], k=7, 最大长度为114.
分析:常见的二分法解决的为index问题,此题可以通过查找k的位置来转化为二分问题。将寻常二分nums[mid] == target,替换为检查func() == k来实现。
public int woodCut(int[] L, int k) {
if (L == null || L.length == 0) {
return 0;
}
int max = 0; //the longest length
for (int i = 0; i < L.length; i++) {
max = Math.max(max,L[i]);
}
int start = 1;
int end = max;
while (start + 1 < end) {
int mid = start + (end - start) / 2;
if (count(L,mid) == k){ //binary search
start = mid;
} else if (count(L,mid) < k) {
end = mid;
} else {
start = mid;
}
}
if (count(L,end) >= k) {
return end;
}
if (count(L,start) >= k) {
return start;
}
return 0;
}
//count the piece of wood when it is the length
private int count (int[] L, int length) {
int num = 0;
for (int i = 0; i < L.length; i++) {
num += L[i] / length;
}
return num;
}