leetcode_915 分割数组

要求

给定一个数组 A，将其划分为两个连续子数组 left 和 right， 使得：

• left 中的每个元素都小于或等于 right 中的每个元素。
• left 和 right 都是非空的。
• left 的长度要尽可能小。 在完成这样的分组后返回 left 的长度。可以保证存在这样的划分方法。

示例 1：

输入：[5,0,3,8,6]
输出：3
解释：left = [5,0,3]，right = [8,6]

示例 2：

输入：[1,1,1,0,6,12]
输出：4
解释：left = [1,1,1,0]，right = [6,12]

提示：

• 2 <= A.length <= 30000
• 0 <= A[i] <= 10^6
• 可以保证至少有一种方法能够按题目所描述的那样对 A 进行划分。

核心代码

class Solution:
    def partitionDisjoint(self, nums: List[int]) -> int:
        l = 1
        lm = nums[0]
        maxItem = nums[0]
        n = len(nums)
        for i in range(l,n):
            if nums[i] < lm:
                l = i + 1
                lm = maxItem
            if nums[i] > maxItem:
                maxItem = nums[i]
        return l

解题思路：假设left的长度为1，并且该元素设为左边部分的最大值lm，那么right就为剩下的n-1。我们可以对剩下的n-1遍历，遍历时假如当前元素小于lm（不满足条件1），说明这个元素一定属于left，那么就要把“刀”移到当前位置。但是还要注意，移动“刀”位置后，left部分的最大值lm可能会发生变化，因此需要在遍历时记录最大元素，以便及时更新lm。