leetcode-1438. 绝对差不超过限制的最长连续子数组

题目

给你一个整数数组 nums ，和一个表示限制的整数 limit，请你返回最长连续子数组的长度，该子数组中的任意两个元素之间的绝对差必须小于或者等于 limit 。

如果不存在满足条件的子数组，则返回 0 。

示例 1：

输入：nums = [8,2,4,7], limit = 4
输出：2 
解释：所有子数组如下：
[8] 最大绝对差 |8-8| = 0 <= 4.
[8,2] 最大绝对差 |8-2| = 6 > 4. 
[8,2,4] 最大绝对差 |8-2| = 6 > 4.
[8,2,4,7] 最大绝对差 |8-2| = 6 > 4.
[2] 最大绝对差 |2-2| = 0 <= 4.
[2,4] 最大绝对差 |2-4| = 2 <= 4.
[2,4,7] 最大绝对差 |2-7| = 5 > 4.
[4] 最大绝对差 |4-4| = 0 <= 4.
[4,7] 最大绝对差 |4-7| = 3 <= 4.
[7] 最大绝对差 |7-7| = 0 <= 4. 
因此，满足题意的最长子数组的长度为 2 。

示例 2：

输入：nums = [10,1,2,4,7,2], limit = 5
输出：4 
解释：满足题意的最长子数组是 [2,4,7,2]，其最大绝对差 |2-7| = 5 <= 5 。

示例 3：

输入：nums = [4,2,2,2,4,4,2,2], limit = 0
输出：3

提示：

1 <= nums.length <= 10^5
1 <= nums[i] <= 10^9
0 <= limit <= 10^9

解题思路

滑动窗口+单调队列

首先，找连续最长子数组的长度，考虑用滑动窗口。
其次，需要求最大值和最小值之差，所以需要维护一个保存最大值的栈、最小值的队列，思路和滑动窗口最大值基本一致。

当滑动窗口向后滑的时候，需要判断滑出去的数字，是否在队列中，如果在队列中，则把队列里的元素出队即可。

代码

class Solution:
    def longestSubarray(self, nums: List[int], limit: int) -> int:
        min_deque, max_deque = collections.deque([]), collections.deque([])
        window_start = 0
        length = 0
        for index, num in enumerate(nums):
            while min_deque and num > nums[min_deque[-1]]:
                min_deque.pop()
            min_deque.append(index)
            while max_deque and num < nums[max_deque[-1]]:
                max_deque.pop()
            max_deque.append(index)
            while min_deque and max_deque and nums[min_deque[0]] - nums[max_deque[0]] > limit:
                if min_deque[0] == window_start:
                    min_deque.popleft()
                if max_deque[0] == window_start:
                    max_deque.popleft()
                window_start += 1
            length = max(length, index - window_start + 1)
        return length