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

发布时间 2023-04-02 10:13:18作者: Co3y

力扣题目链接

给你一个整数数组 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
 1 class Solution {
 2     public int longestSubarray(int[] nums, int limit) {
 3         //TreeMap是有序的!
 4         TreeMap<Integer, Integer> window = new TreeMap<>();
 5         int n = nums.length;
 6         int left = 0, right = 0;
 7         int res = 0;
 8         while (right < n) {
 9             int r = nums[right];
10             right++;
11             window.put(r, window.getOrDefault(r, 0) + 1);
12             //寻找最长模板(while为窗口不满足条件,结果在外部更新)
13             //lastKey()和firstKey()返回该TreeMap的最后一个(最大的)/第一个(最小的)映射的key
14             while (window.lastKey() - window.firstKey() > limit) {
15                 int l = nums[left];
16                 left++;
17                 window.put(l, window.get(l) - 1);
18                 //当value==0时,需要将对应key remove不然会导致跳不出循环,越界。
19                 if (window.get(l) == 0)
20                     window.remove(l);
21             }
22             //right-left即为窗口有效长度,因为我维持的是[left, right)的窗口,具体原因在总结时说明了
23             res = Math.max(res, right - left);
24         }
25         return res;
26     }
27 }

时间复杂度:O(nlog⁡n),其中 n 是数组长度。向有序集合TreeMap中添加或删除元素都是 O(log⁡n) 的时间复杂度。每个元素最多被添加与删除一次。

空间复杂度:O(n),其中 n 是数组长度。最坏情况下有序集合将和原数组等大。