【leetcode】分割数组的最大值(二分枚举)

给定一个非负整数数组和一个整数 m,你需要将这个数组分成 m 个非空的连续子数组。设计一个算法使得这 m 个子数组各自和的最大值最小。

注意:
数组长度 n 满足以下条件:

1 ≤ n ≤ 1000
1 ≤ m ≤ min(50, n)
示例:

输入:
nums = [7,2,5,10,8]
m = 2

输出:
18

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

链接:https://leetcode-cn.com/problems/split-array-largest-sum

思路分析:

二分法:

这道题比较难想到用二分,因为潜在的有序关系不直观。
二分枚举的题,我们首先需要思考3个问题:

  1. 二分枚举的相关判定变量是否具有单调性?
  2. l, r, mid 代表着什么?
  3. 用于缩小区间的判断条件是什么?

对于本题来说,我们需要求 在m次划分下 最小的 子数组各自和最大值。
令 x 为划分数,y为子数组各自和最大值。可以看出,x范围在[1,nums.size],并且x与y单调递减。这里的 x 和 y 就是相关判定变量。
当 x = 1 时,y就是整个nums数组的和,也就是 r 所代表的值。
当 x = nums.size 时,y就是nums中最大元素的值,也就是 l 所代表的值。
通过 (l+r)/2 所得到的 mid 就是枚举值,充当一个假设的已知条件。

现在,我们的已知条件是 m 和 mid. m相当于题目给出的 x = m, mid相当于我们在值域中枚举的一个数 y’. 当 x = m,y可能存在许多值,即 x=m 对应一个 y 的集合。我们的任务是求 y 集合中的最小值。
首先,我们得知道我们的 mid 与 y 的关系,然后才能根据这个关系二分缩小区间,直到 mid = y.
如何知道 mid 与 y的关系呢?前面说到 mid 相当于 y’. 我们通过mid可以求出 x’,然后让 x和x’比较。因为x与y单调递减,所以如果x’ > x 则表示 mid < y,区间需要向右缩小。这样就可以达到我们的目的了。

二分枚举的原理说明白了,现在简要说一下步骤。

  1. 找到二分枚举的相关判定变量 x 和 y. y是题目要求的值,此题代表子数组各自和最大值。x是可以影响到 y 的值,此题代表划分数。
  2. 判断 x 和 y 之间的关系。可以发现,x 和 y 单调递减。
  3. 通过 x 的范围算出 y 的范围,并取边界值 l 和 r 作为枚举范围。
  4. 算出 mid,即 y’. 再通过 y’ 算出 x’,然后通过比较 x 和 x’ 的关系得到 y 和 y’ 的关系,进而缩小枚举区间。即通过 mid 算出 cnt,然后令 cnt 和 m 比较。
  5. 根据题意,cnt > m,区间缩小到右半部分;cnt < m,区间缩小到左半部分;cnt = m,区间缩小到左半部分(求y集合的最小值)。
  6. 最后,l (或 r) 的值就是所求值。
class Solution {
public:
    int splitArray(vector<int>& nums, int m) {
        int n = nums.size();
        long long l = nums[0], r = 0;
        for(int i = 0;i < n;i++)
        {
            r += nums[i];
            l = nums[i]>l?nums[i]:l;
        }
        while(l < r)
        {
            long long mid = (l+r)/2;	//y'
            int cnt = 1;	//x'
            long long tmp = 0;
            for(int i = 0;i < n;i++)
            {
                tmp += nums[i];
                if(tmp > mid)
                {
                    tmp = nums[i];
                    cnt++;	//通过y'求出x'
                }
            }
            if(cnt > m) l = mid+1;	//通过比较x与x'来缩小区间
            else r = mid;	
        }
        return l;
    }
};
题目描述:给定一个非负整数数组nums和一个整数m,你需要将这个数组分成m个非空的连续子数组。设计一个算法使得这m个子数组中的最大和最小。 解题思路: 这是一个典型的二分搜索题目,可以使用二分查找来解决。 1. 首先确定二分的左右边界。左边界为数组中最大的值,右边界为数组中所有元素之和。 2. 在二分搜索的过程中,计算出分割数组的组数count,需要使用当前的中间值来进行判断。若当前的中间值不够分割成m个子数组,则说明mid值偏小,将左边界更新为mid+1;否则,说明mid值偏大,将右边界更新为mid。 3. 当左边界小于等于右边界时,循环终止,此时的左边界即为所求的结果。 具体步骤: 1. 遍历数组,找到数组中的最大值,并计算数组的总和。 2. 利用二分查找搜索左右边界,从左边界到右边界中间的值为mid。 3. 判断当前的mid值是否满足题目要求,若满足则更新右边界为mid-1; 4. 否则,更新左边界为mid+1。 5. 当左边界大于右边界时,循环终止,返回左边界即为所求的结果。 代码实现: ```python class Solution: def splitArray(self, nums: List[int], m: int) -> int: left = max(nums) right = sum(nums) while left <= right: mid = (left + right) // 2 count = 1 total = 0 for num in nums: total += num if total > mid: total = num count += 1 if count > m: left = mid + 1 else: right = mid - 1 return left ``` 时间复杂度分析:二分搜索的时间复杂度为O(logN),其中N为数组的总和,而遍历数组的时间复杂度为O(N),因此总的时间复杂度为O(NlogN)。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值