绝对差不超过限制的最长连续子数组——滑动窗口的两种做法

绝对差不超过限制的最长连续子数组,题目地址

一、思路

  1. 使用滑动窗口保持符合条件的子数组,记录最长长度
  2. 需要两个数据:滑动窗口中的最大值,最小值
  3. 需要对滑入窗口的数据记录,滑出的数据删除,并且更新出最大值和最小值

二、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;
	}
};

猜你喜欢

转载自blog.csdn.net/weixin_45605341/article/details/108679244