leetcode1231 分享巧克力
作者:互联网
https://leetcode-cn.com/problems/divide-chocolate/
将数组分为K+1份。最合理的方式就是均分。
遍历数组,逐渐的累加,当累加的值大于等于arvg,清空当前的累加值,继续进行下一次的累加,并且当前记录为划分一块。当遍历结束,或者还未结束时,记录的划分的块数已经达标。那么当前均值arvg满足要求。否则就将arvg–,重复上述操作。
public class demo02 {
public int maxMin(int[] arr, int k){
int sum = 0;
for(int i : arr){
sum += i;
}
int avg = sum / (k+1);
while(avg > 0){
int cur = 0;
int curSum = 0;
for(int i : arr){
curSum += i;
if(curSum >= avg){
if(cur++ >= k){
return avg;
}
curSum = 0;
}
}
avg--;
}
return avg;
}
}
思路已经很清晰了,那么如何提高效率呢。此处avg是通过一步步的减去逐步逼近,可以改为二分查找。
public class demo02 {
public int maxMin(int[] arr, int k){
int ans = 0;
int left = 0;
int right = (int) 1e9;
while(left <= right){
int mid = (left + right) >> 1;
if(check(arr,k+1,mid)){
left = mid + 1;
ans = Math.max(ans,mid);
}else{
right = mid-1;
}
}
return ans;
}
public boolean check(int[] arr, int len, int avg){
int cur = 0;
int curSum = 0;
for(int a : arr){
curSum += a;
if(curSum >= avg){
cur++;
curSum = 0;
}
}
return cur > len;
}
}
标签:巧克力,avg,cur,int,curSum,leetcode1231,arr,mid,分享 来源: https://blog.csdn.net/xiaobailaji/article/details/116646803