915. 分割数组

2022/10/24 leetcode 共 588 字,约 2 分钟

title

problem link

给定一个数组 nums ,将其划分为两个连续子数组 left 和 right, 使得:

  • left 中的每个元素都小于或等于 right 中的每个元素。
  • left 和 right 都是非空的。
  • left 的长度要尽可能小。

在完成这样的分组后返回 left 的 长度 。

用例可以保证存在这样的划分方法。

示例 1:

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

示例 2:

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

提示:

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

solution

下面为一次遍历,使用后悔法。

class Solution:
    def partitionDisjoint(self, nums: List[int]) -> int:
        n = len(nums)
        leftpos = 0
        leftmax = nums[0]
        currentmax = nums[0]
        for i,v in enumerate(nums):
            currentmax = max(currentmax, v)
            if v < leftmax:
                leftpos = i
                leftmax = currentmax
        return leftpos + 1

Search

    Table of Contents