581. Shortest Unsorted Continuous Subarray

Description

Given an integer array nums, you need to find one continuous subarray that if you only sort this subarray in ascending order, then the whole array will be sorted in ascending order.

Return the shortest such subarray and output its length.

 

Example 1:

Input: nums = [2,6,4,8,10,9,15]
Output: 5
Explanation: You need to sort [6, 4, 8, 10, 9] in ascending order to make the whole array sorted in ascending order.

Example 2:

Input: nums = [1,2,3,4]
Output: 0

Example 3:

Input: nums = [1]
Output: 0

 

Constraints:

• 1 <= nums.length <= 104
• -105 <= nums[i] <= 105

 

Follow up: Can you solve it in O(n) time complexity?

Solution

shortest-unsorted-continuous-subarray.py

class Solution:
    def findUnsortedSubarray(self, nums: List[int]) -> int:
        n = len(nums)

        right, prefixMax = -1, nums[0]

        for i in range(1, n):
            if nums[i] < prefixMax:
                right = i
            else:
                prefixMax = nums[i]

        if right == -1: return 0

        left, suffixSmall = n - 1, nums[-1]

        for i in range(n - 2, -1, -1):
            if nums[i] > suffixSmall:
                left = i
            else:
                suffixSmall = nums[i]

        return right - left + 1