1.下一个更大元素 II
由于其为首尾相接的特性,那么我们只需要多遍历一次数组就可以。为了不增加时间复杂度,我们直接取余来操作。
class Solution {
public:
vector<int> nextGreaterElements(vector<int>& nums) {
stack<int>s;
vector<int>ret(nums.size(),-1);
for(int i=0;i<nums.size()*2;i++)
{
while(s.size()!=0&&nums[s.top()%nums.size()]<nums[i%nums.size()])
{
ret[s.top()%nums.size()]=nums[i%nums.size()];
s.pop();
}
s.push(i);
}
return ret;
}
};
2.接雨水
1.先定义一个栈,该承接之前的思路。不过此时我们还需要知道这只能得到比当前值要大数
2.那其实我们要找的就是当前位置的前后第一个比当前数大的数。
那单调递增栈当前位置的值比当前栈顶的值大的时,说明此时的栈顶就是接雨水的凹槽处,那么此时我们需要先计算左右两边能接水的最小高度,min(height[st.top()], height[i]) - height[mid];此时就是左右的较小一边为高度所能接到的高度,那么两边的宽度为i - st.top() - 1,最后得到的结果就是当前以最小高度为基准能装入的雨量
class Solution {
public:
int trap(vector<int>& height) {
if (height.size() <= 2) return 0;
int ret = 0;
stack<int> st;
for (int i = 0; i < height.size(); i++)
{
if (st.empty())
{
st.push(i);
continue;
}
if (height[st.top()] > height[i])
st.push(i);
else if (height[st.top()] == height[i])
st.push(i);
else
{
while (!st.empty() && height[st.top()] < height[i])
{
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;
ret += h * w;
}
}
st.push(i);
}
}
return ret;
}
};
3.柱状图中最大的矩形
1.本题与上一题的本质区别在于我们要找到的值为左右第一个比当前值要小的值
2.那么思路为:我们得到某一个位置,我们需要知道它左右两边的第一个小值,左右两个值的中间范围为宽度,高度为当前值,那么高度乘以宽度就是我们需要的矩形大小
3.那么我们需要的是一个单调递减栈,当前值要比栈顶值小时,说明此时的栈顶就是中间值,它右边范围为当前值下标,它左边范围为栈中后一个元素,因为单调递减栈中栈顶是最大的,也就是说栈顶比栈里的元素要大。那么我们分别取出栈顶为中间值,出栈后的栈顶为左范围,当前下标为右边值,那么我们就可以求出面积大小。每次的去面积都要和上一次存储的值进行比较大小,我们需要保留较大的那一个面积
4.此外数组的首位都需要insert一个0,因为所谓的求面积是基于栈中一点存在元素,那么设想一下,如果整个数据就是单调的,那么我们就没有存储左边范围或者右边范围。那么就不会得到我们想求出的值。
class Solution {
public:
int largestRectangleArea(vector<int>& heights) {
heights.insert(heights.begin(), 0);
heights.push_back(0);
int ret = 0;
stack<int> st;
for (int i = 0; i < heights.size(); i++)
{
while(!st.empty()&&heights[i]<heights[st.top()])
{
int mid = st.top();
st.pop();
int left = st.top();
int right = i;
int h = heights[mid];
int w = right-left-1;
ret=max(ret,h*w);
}
st.push(i);
}
return ret;
}
};