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
方法一:滑动窗口 + 单调栈
解题思路
- 用双指针
[left, right]
模拟滑动窗口,保持窗口内的最大差值不大于limit
。每次计算窗口的大小,最大值即为结果。 - 当有新元素加入时,可能会破坏滑窗,使最大差值大于
limit
,但是我们并不知道“破坏性有多强”,即需要移动left
指针到什么位置才能保持滑动的稳定性(最大差值不大于limit
)。 - 为此,在窗口内维护两个 单调栈。
large
栈内元素单调递减;small
栈内单调递增,最大差值即为large.first - small.first
。 新增的元素会同时加入这两个栈,根据两个栈首元素来使得最大差值不会超过limit
。
参考代码
public int longestSubarray(int[] nums, int limit) {
Deque<Integer> large = new LinkedList<>();
Deque<Integer> small = new LinkedList<>();
int n = nums.length;
int ret = 0;
int left = 0, right = 0;
while (right < n) {
while (!large.isEmpty() && large.peekLast() < nums[right]) {
large.pollLast();
}
while (!small.isEmpty() && small.peekLast() > nums[right]) {
small.pollLast();
}
large.offerLast(nums[right]);
small.offerLast(nums[right]);
while (large.peekFirst() - small.peekFirst() > limit) {
if (large.peekFirst() == nums[left]) {
large.pollFirst();
}
if (small.peekFirst() == nums[left]) {
small.pollFirst();
}
left++;
}
right++;
ret = Math.max(ret, right - left);
}
return ret;
}
执行结果