Description
给你一个整数数组 nums ,和一个表示限制的整数 limit,请你返回最长连续子数组的长度,该子数组中的任意两个元素之间的绝对差必须小于或者等于 limit 。
如果不存在满足条件的子数组,则返回 0 。
Solution
首先发现显然可以用双指针来计算答案,重点在于如何维护当前段内的最大最小值
动态维护当前窗口内的最大值和最小值,可以采用单调队列
时间复杂度
O
(
N
)
O(N)
O(N)
空间复杂度
O
(
N
)
O(N)
O(N)
Code
class Solution {
public:
static const int maxn = 1e5 + 7;
static const int inf = 1e9 + 7;
int head1, tail1, que1[maxn];// max
int head2, tail2, que2[maxn];// min
void init() {head1 = tail1 = head2 = tail2 = 0;}
inline int mx() {
if(head1 < tail1) return que1[head1];return -inf;
}
inline int mi() {
if(head2 < tail2) return que2[head2];return inf;
}
inline void insert(int x) {
while(head1 < tail1 && x > que1[tail1-1]) tail1--;
que1[tail1++] = x;
while(head2 < tail2 && x < que2[tail2-1]) tail2--;
que2[tail2++] = x;
}
inline void pop_front(int x) {
if(head1 < tail1 && que1[head1] == x) head1++;
if(head2 < tail2 && que2[head2] == x) head2++;
}
int longestSubarray(vector<int>& nums, int limit) {
init();
int n = nums.size();if(n == 0) return 0;
int i = 0,j = 0, res = 1;
insert(nums[0]);
while(i < n) {
while(j + 1 < n) {
int a = max(nums[j+1], mx()), b = min(nums[j+1], mi());
if(a - b <= limit) {
insert(nums[j+1]);j++;
} else break;
}res = max(res, j - i + 1);
pop_front(nums[i++]);
}return res;
}
};