【算法日志】单调栈: 单调栈简介及其应用

代码随想录刷题60Day


目录

单调栈简介

单调栈的应用

下次更高温

下一个更大元素1

下一个更大元素2

接雨水

柱状图中最大矩形


单调栈简介

单调栈(Monotonic Stack)是一种特殊的栈数据结构,它满足元素的单调性,这种单调性需要自己建立和维护。单调栈分为单调递增栈和单调递减栈两种类型。

单调递增栈的特点是栈内元素从栈底到栈顶依次递增,而单调递减栈则是栈内元素从栈底到栈顶依次递减。

单调栈的主要应用是解决一些与找到元素的下一个更大或更小相关的问题。它通过维护一个递增或递减的栈,可以在常数时间内找到每个元素的下一个更大或更小的元素。

单调栈的基本操作包括:

入栈:将元素压入栈顶,同时保持栈的单调性。

出栈:从栈顶移除元素。

查找:检查栈顶元素,获取当前元素的下一个更大或更小的元素。

单调栈的应用

下次更高温

【算法日志】单调栈: 单调栈简介及其应用_第1张图片

    vector dailyTemperatures(vector& temperatures) 
	{
		int size = temperatures.size();
		vector result(size, 0);
		stack stack;
		stack.push(0);
		for (int i = 1; i < size; ++i)
		{
			int j = stack.top();
			if (temperatures[i] > temperatures[j])
			{
				while (!stack.empty() && temperatures[i] > temperatures[j])
				{
					result[j] = i - j;
					stack.pop();
					if (!stack.empty())j = stack.top();
				}
			}
			stack.push(i);
		}
		return result;
	}

下一个更大元素1

【算法日志】单调栈: 单调栈简介及其应用_第2张图片

	vector nextGreaterElement(vector& nums1, vector& nums2) 
	{
		int size1 = nums1.size();
		int size2 = nums2.size();
		unordered_map umap;
		stack stack;
		vector result;
		stack.push(0);
		for (int i = 1; i < size2; ++i)
		{
			int j = stack.top();
			if (nums2[j] < nums2[i])
			{
				while (!stack.empty() && nums2[j] < nums2[i])
				{
					umap.insert(pair(nums2[j], nums2[i]));
					stack.pop();
					if (!stack.empty())j = stack.top();
				}
			}
			stack.push(i);
		}
		for (int i = 0; i < size1; ++i)
		{
			if (umap.find(nums1[i]) != umap.end())
				result.push_back(umap[nums1[i]]);
			else
				result.push_back(-1);
		}
		return result;
	}

下一个更大元素2

【算法日志】单调栈: 单调栈简介及其应用_第3张图片

	vector nextGreaterElements(vector& nums) 
	{
		int size = nums.size();
		vector dp(size, -1);
		stack mystack;
		mystack.push(0);
		for (int i = 1; i < size; ++i)
		{
			if (nums[i] > nums[mystack.top()])
			{
				while (!mystack.empty() && nums[i] > nums[mystack.top()])
				{
					dp[mystack.top()] = nums[i];
					if (!mystack.empty())mystack.pop();					
				}
			}
			mystack.push(i);
		}
		for (int i = 0; i < size; ++i)
		{
			if (nums[i] > nums[mystack.top()])
			{
				while (!mystack.empty() && nums[i] > nums[mystack.top()])
				{
					dp[mystack.top()] = nums[i];
					if (!mystack.empty())mystack.pop();					
				}
			}
			mystack.push(i);
		}
		return dp;
	}

接雨水

【算法日志】单调栈: 单调栈简介及其应用_第4张图片

	int trap(vector& h)
	{
		if (h.size() < 3)return 0;
		stack mystack;
		int result = 0;
		mystack.push(0);
		for (int i = 1; i < h.size(); ++i)
		{
			if (h[mystack.top()] > h[i])
				mystack.push(i);
			else if (h[mystack.top()] < h[i])
			{
				while (!mystack.empty() && h[mystack.top()] < h[i])
				{
					int mid = mystack.top();
					mystack.pop();
					if (!mystack.empty())
						result += (min(h[mystack.top()], h[i]) - h[mid]) * (i - mystack.top() - 1);				
				}
				if (!mystack.empty() && h[mystack.top()] == h[i])
					mystack.pop();
				mystack.push(i);
			}
			else
			{
				mystack.pop();
				mystack.push(i);
			}
		}
		return result;
	}

柱状图中最大矩形

【算法日志】单调栈: 单调栈简介及其应用_第5张图片

	int largestRectangleArea(vector& h) 
	{
		h.push_back(0);
		int size = h.size();
		stack mystack;
		int result = h[0];
		mystack.push(0);
		for (int i = 1; i < size; ++i)
		{
			if (h[i] == h[mystack.top()])
			{
				mystack.pop();
			}
			else if (h[i] < h[mystack.top()])
			{
				int mid, left;
				while (!mystack.empty() && h[i] < h[mystack.top()])
				{
					mid = mystack.top();
					mystack.pop();
					if (!mystack.empty())
						left = mystack.top();
					else
						left = -1;
					result = max(result, (i - left - 1) * h[mid]);
				}
			}
			mystack.push(i);
		}
		return result;
	}

你可能感兴趣的:(数据结构,开发语言,c++,leetcode)