编程语言
首页 > 编程语言> > 【LeetCode&python】915. 分割数组

【LeetCode&python】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]

思路&代码

  1. 第一次很简单的设想拿左面数组部分的最大值和右面数组部分的最小值比较,如果最大值小于或等于最小值,那么就可以输出下标+1作为位置。
    时间复杂度是包含一次循环和一次max或min调用,为O(n^2),很自然地超时了
class Solution:
    def partitionDisjoint(self, A: List[int]) -> int:
        max_A=A[0]
        for i in range(len(A)):
            if (max_A < A[i]):
                max_A = A[i]
            if(max_A<=min(A[i+1::])):
                return i+1
  1. 优化一下,仔细想想,会发现其实如果遍历数组的时候,分为两个状态其实就可以把复杂度变为O(n)了,即遍历一遍就可以得到结果。
class Solution:
    def partitionDisjoint(self, A: List[int]) -> int:
        candidate=A[0] # 候选数
        max_left=A[0] # 左面已经遍历过的最大值
        record=0 # 记录下左面全部小于候选数的最后的位置
        flag=0 # 标志位,用于检测是否遇到大于候选数之后又遇到小于候选数的情况
        for i in range(len(A)):
            if(flag==0):
                if(candidate>A[i]):
                    record=i
                if (candidate < A[i]):
                    flag=1
                if(max_left<=A[i]):
                    max_left=A[i]
            else:
                # 遇到大于候选数之后又遇到小于候选数的情况
                if (max_left <= A[i]):
                    max_left = A[i]

                if (candidate <= A[i]):
                    pass
                else:
                    candidate=max_left
                    flag=0
                    record=i
        return record+1

标签:候选,python,max,record,左面,数组,915,LeetCode,left
来源: https://www.cnblogs.com/heyjjjjj/p/13804896.html