绝对差不超过限制的最长连续子数组,题目地址
一、思路
- 使用滑动窗口保持符合条件的子数组,记录最长长度
- 需要两个数据:滑动窗口中的最大值,最小值
- 需要对滑入窗口的数据记录,滑出的数据删除,并且更新出最大值和最小值
二、map+滑动窗口
我们可以使用map来记录,因为map内部自建一颗红黑树(一 种非严格意义上的平衡二叉树),这颗树具有对数据自动排序的功能,所以在map内部所有的数据都是有序的,所有正好可以找出滑动数组中的最大最小值,以此来更新。
class Solution {
public:
int longestSubarray(vector<int>& nums, int limit)
{
map < int, int > m;
int ans = 0;
int j = 0;
for (int i = 0; i < nums.size(); i++)
{
m[nums[i]]++;
while (m.rbegin()->first - m.begin()->first > limit)
{
m[nums[j]]--;//j位置的数划出
if (m[nums[j]] == 0)//数量为0直接删除该键值对
m.erase(nums[j]);
j++;
}
ans = max(ans, i - j + 1);//更新
}
return ans;
}
};
三、deque(队列)
两个单调队列维护最大值,最小值,队头分别表示滑动窗口中的最小值、最大值。当 abs(q1.front() - q2.front()) > limit时,窗口缩小,j++。
这样的话复杂度只有O(n)
class Solution {
public:
int longestSubarray(vector<int>& nums, int limit)
{
int ans = 0;
deque<int> q1, q2; //双端队列来维护单调队列
for (int i = 0, j = 0; i < nums.size(); i++)
{
//队头分别表示滑动窗口中的最小值、最大值。
while (!q1.empty() && q1.back() > nums[i]) q1.pop_back();//维护单调增队列
q1.push_back(nums[i]);
while (!q2.empty() && q2.back() < nums[i]) q2.pop_back();//维护单调减队列
q2.push_back(nums[i]);
//检测当前窗口是否满足条件
while (!q1.empty() && !q2.empty() && q2.front() - q1.front() > limit)
{
if (nums[j] == q1.front()) q1.pop_front(); //从队头出队
if (nums[j] == q2.front()) q2.pop_front(); //从队头出队
j++;
}
ans = max(ans, i - j + 1);
}
return ans;
}
};