其他分享
首页 > 其他分享> > leetcode1231 分享巧克力

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