数据结构与算法笔记-二分查找-力扣410

最新推荐文章于 2024-04-16 17:27:15 发布

Shaeyln Ma

最新推荐文章于 2024-04-16 17:27:15 发布

阅读量98

点赞数

分类专栏：算法刷题笔记文章标签： leetcode 数据结构算法 java

本文链接：https://blog.csdn.net/Shaeyln2136/article/details/112388002

版权

算法刷题笔记专栏收录该内容

4 篇文章 0 订阅

订阅专栏

题目：分割数组的最大值

给定一个非负整数数组和一个整数 m，你需要将这个数组分成 m 个非空的连续子数组。设计一个算法使得这 m 个子数组各自和的最大值最小
输入:
nums = [7,2,5,10,8]
m = 2

输出:
18

解释:
一共有四种方法将nums分割为2个子数组。
其中最好的方式是将其分为[7,2,5] 和 [10,8]，
因为此时这两个子数组各自的和的最大值为18，在所有情况中最小。

思路

用子数组各自和的最大值作为标定进行二分查找。

边界的确定，考虑两种极端情况：

每个子数组均只有一个元素，那么各自和的最大值就是整个数组的最大值，该最大值就是二分查找的左边界
整个数组只被划分为一个子数组，那么各自和的最大值就是整个数组的和，该和就是二分查找的右边界

区域缩减情况： int mid=(left+right)/2

当以mid为最大值界限，不能满足划分条件时，（即超过了要求的只能划分为m个数组），将子数组各自和的最大值调大，left=mid+1
当以mid为最大值界限，能满足划分条件时，（即划分为m个数组绰绰有余），将子数组各自和的最大值调小，right=mid

构造以当前mid为子数组最大值能否满足划分要求的方法

代码

class Solution {
    public int splitArray(int[] nums, int m) {
//二分法，左边界为数组的最大值，右边界为数组的和
    int max=0;
    int sum=0;
    for(int num:nums){
        sum+=num;
        max=Math.max(max,num);
    }
    int left=max;
    int right=sum;
    while(left<right){
        int mid=(left+right)/2;
        if(isOk(nums,mid,m)){//说明当前选取的最大值可以满足分割条件，调小子数组各自和最大值
         right=mid;
        
        }else{
         left=mid+1;  
        }
    }
    return right;
    }
    private boolean isOk(int []nums,int maxIntervalSum,int m){
        int curIntervalSum=0;//当前区间的和
        for(int num:nums){
            if(curIntervalSum+num>maxIntervalSum){
                curIntervalSum=0;
                m--;
            }
            curIntervalSum+=num;
        }
        return m>0;
    }
}