算法学习——单调栈

lei453913232

已于 2024-07-17 17:41:05 修改

阅读量232

点赞数 1

文章标签：算法数据结构

于 2024-07-16 17:12:37 首次发布

本文链接：https://blog.csdn.net/lei453913232/article/details/140467410

版权

原理：栈里保存的数值保持单调性，用于保存之前遍历过的、待使用的元素

一、每日温度（经典裸单调栈）

原题力扣739 . - 力扣（LeetCode）

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

二、下一个更大元素I（哈希+单调栈）

原题力扣496 . - 力扣（LeetCode）

vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2)
{
    unordered_map<int, int> mp;
    for (int i = 0; i < nums1.size(); i++)
    {
        mp[nums1[i]] = i;
    }
    vector<int> ans(nums1.size(), -1);
    stack<int> st;
    st.push(0);
    for (int i = 1; i < nums2.size(); i++)
    {
        if (nums2[i] <= nums2[st.top()])
        {
            st.push(i);
        }
        else
        {
            while (!st.empty() && nums2[i] > nums2[st.top()])
            {
                if (mp.count(nums2[st.top()]))
                {
                    ans[mp[nums2[st.top()]]] = nums2[i];
                    st.pop();
                }
                else
                {
                    st.pop();
                }
            }
            st.push(i);
        }
    }
    return ans;
}

三、下一个更大元素II（环形数组）

原题力扣503 . - 力扣（LeetCode）

通过取模模拟成环过程

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

四、接雨水（找两边都更高的）

原题力扣42 . - 力扣（LeetCode）

int trap(vector<int>& height)
{
	int ans = 0;
	stack<int> st;
	st.push(0);
	for (int i = 1; i < height.size(); i++)
	{
		if (height[i] < height[st.top()])
		{
			st.push(i);
		}
		else if (height[i] == height[st.top()]) // 等于时，没有高度差，不影响计算结果，可抛弃
		{
			st.pop();
			st.push(i);
		}
		else
		{
			while (!st.empty() && height[i] > height[st.top()]) // 找到了凹陷下去的柱子，现需寻找其左右两边比它高的柱子
			{
				int mid = st.top(); // 凹陷下去的柱子位置
				st.pop();
				if (!st.empty())
				{
					// 高度 = min(左边第一个比凹陷柱子高的柱子高度，右边第一个比凹陷柱子高的柱子高度) - 凹陷柱子高度
					int h = min(height[st.top()], height[i]) - height[mid];
					// 宽度 = 右边柱子位置 - 左边柱子位置 - 1
					int w = i - st.top() - 1;
					// 面积 = 高度 * 宽度（横向计算，累加）
					ans += h * w;
				}
			}
			st.push(i);
		}
	}
	return ans;
}

五、柱状图中最大的矩形（找两边都更低的）

原题力扣84 . - 力扣（LeetCode）

int largestRectangleArea(vector<int>& heights)
{
	int ans = 0;
	// 首尾加 0 防止死循环
	heights.insert(heights.begin(), 0);
	heights.push_back(0);
	stack<int> st;
	st.push(0);
	for (int i = 1; i < heights.size(); i++)
	{
		if (heights[i] >= heights[st.top()])
		{
			st.push(i);
		}
		else
		{
			// 找到左边和右边第一个比当前凸出柱子低的
			while (!st.empty() && heights[i] < heights[st.top()])
			{
				int mid = st.top();
				st.pop();
				if (!st.empty())
				{
					int h = heights[mid]; // 高即为 当前凸出柱子的高度
					int w = i - st.top() - 1; // 宽即为 左右两边柱子位置差 - 1
					ans = max(ans, h * w);
				}
			}
			st.push(i);
		}
	}
	return ans;
}