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:
        ans = 0
        q_min = deque()
        q_max = deque()
        left = -1

        for i, x in enumerate(nums):
            while q_max and x > nums[q_max[-1]]:
                q_max.pop()
            q_max.append(i)

            while q_min and x < nums[q_min[-1]]:
                q_min.pop()
            q_min.append(i)

            while nums[q_max[0]] - nums[q_min[0]] > limit:
                if q_max[0] < q_min[0]:
                    left = q_max.popleft()
                else:
                    left = q_min.popleft()
            
            ans = max(ans, i - left)
        
        return ans

C++ 代码

class Solution {
public:
    int longestSubarray(vector<int>& nums, int limit) {
        deque<int> queMax;
        deque<int> queMin;

        int n = nums.size();
        int left = -1;
        int ans = 0;

        for (int i = 0; i < n; i ++) {
            while (queMax.size() && nums[i] > nums[queMax.back()]) {
                queMax.pop_back();
            }

            while (queMin.size() && nums[i] < nums[queMin.back()]) {
                queMin.pop_back();
            }

            queMax.push_back(i);
            queMin.push_back(i);

            while (nums[queMax.front()] - nums[queMin.front()] > limit) {
                if (queMax.front() < queMin.front()) {
                    left = queMax.front();
                    queMax.pop_front();
                }
                else {
                    left = queMin.front();
                    queMin.pop_front();
                }
            }

            ans = max(ans, i - left);
        }

        return ans;
    }
};

C++ 的 pop_front() 没有返回值。