【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 进行划分。

代码1

思路1

维护两个数组，从左到右，目前遇到的最大元素lmax；从右到左，目前遇到的最小的元素rmin。
从1到len(A)-1，遍历i。以i为分割界限，小于i的为左边数组，大于等于i的为右边数组。只要满足lmax[i-1] <= rmin[i]即可

class Solution:
    def partitionDisjoint(self, A: List[int]) -> int:
        lmax = [0 for i in range(len(A))]
        lmax[0] = A[0]
        for i in range(1,len(A)):
            lmax[i] = max(lmax[i-1], A[i])

        rmin = [0 for i in range(len(A))]
        rmin[-1] = A[-1]
        for i in range(len(A)-2, -1, -1):
            rmin[i] = min(rmin[i+1], A[i])

        for i in range(1,len(A)):
            if(lmax[i-1] <= rmin[i]):
                return i

代码2

class Solution:
    def partitionDisjoint(self, A: List[int]) -> int:
        rmin = min(A)
        lmax = A[0]
        for i in range(1,len(A)):
            if(lmax <= rmin):
                return i
            else:
                if(A[i] > lmax):
                    lmax = A[i]
                if(A[i] <= rmin):
                    rmin = min(A[i+1:])