503.下一个更大元素II
题目链接
力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台
求解思路
重点在如何处理循环数组。
方案一:
直接将两个数组拼接在一起,然后使用单调栈求下一个最大值。
方案二:
在遍历的过程中模拟走两遍nums(详见代码)。
代码
class Solution {
public:vector<int> nextGreaterElements(vector<int>& nums) {vector<int> result(nums.size(), -1);if (nums.size() == 0) return result;stack<int> st;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 if (nums[i % nums.size()] == nums[st.top()]) st.push(i % nums.size());else{while (!st.empty() && nums[i % nums.size()] > nums[st.top()]){result[st.top()] = nums[i % nums.size()];st.pop();}st.push(i % nums.size());}}return result;}
};
42. 接雨水
题目链接
力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台
求解思路
对于一个元素,寻找它的右边和左边的第一个最大元素。
本题的几个重点:
1.单调栈是按行方向来计算雨水的
2.单调栈的元素顺序(从栈头到栈底)
应该是从小到大的顺序。一旦发现添加的柱子高度大于栈头元素了,此时就出现凹槽了,栈头元素就是凹槽底部的柱子,栈头第二个元素就是凹槽左边的柱子,而添加的元素就是凹槽右边的柱子。
3.遇到相同高度的柱子的处理
遇到相同的元素,更新栈内下标,就是将栈里元素(旧下标)弹出,将新元素(新下标)加入栈中。
4.栈内保存的数值
存放下标。
单调栈的处理逻辑:
三种情况
1.当前遍历的元素(柱子)高度小于栈顶元素的高度
把这个元素加入栈中。
2.当前遍历的元素(柱子)高度等于栈顶元素的高度
更新栈顶元素。
3.当前遍历的元素(柱子)高度大于栈顶元素的高度
此时出现凹槽。
取栈顶元素,将栈顶元素弹出,这个就是凹槽的底部,也就是中间位置,下标记为mid,对应的高度为height[mid]。
此时的栈顶元素st.top(),就是凹槽的左边位置,下标为st.top(),对应的高度为height[st.top()](就是图中的高度2)。
当前遍历的元素i,就是凹槽右边的位置,下标为i,对应的高度为height[i](就是图中的高度3)。
雨水高度是(凹槽左边和右边高度的较小值)-(凹槽底部高度),雨水宽度是凹槽右边下标-凹槽左边下标-1,雨水的体积是高度*宽度。
代码
class Solution {
public:int trap(vector<int>& height) {if (height.size() <= 2) return 0;stack<int> st;st.push(0);int sum = 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()){int h = min(height[st.top()], height[i]) - height[mid];int w = i - st.top() - 1;sum += h * w;}}st.push(i);}}return sum;}
};