42. 接雨水
利用单调栈寻找每一个元素右边比它大的元素,同时在单调栈中的下一个元素就是在它左边比它大的第一个元素,此时便形成了凹槽。
class Solution {
public:
int trap(vector<int>& height) {
int result = 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 middle = st.top();
st.pop();
if (!st.empty()) {
int h = min(height[st.top()], height[i]) - height[middle];
int width = i - st.top() - 1;
result += h * width;
}
}
st.push(i);
}
}
return result;
}
};
84.柱状图中最大的矩形
每一个元素向左找到第一个比它小的元素,向右找到第一个比它小的元素。
class Solution {
public:
int largestRectangleArea(vector<int>& heights) {
int result = 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 if (heights[i] == heights[st.top()]) {
st.pop();
st.push(i);
} else {
while (!st.empty() && heights[i] < heights[st.top()]) {
int mid = st.top();
st.pop();
if (!st.empty()) {
int left = st.top();
int right = i;
result = max(result, heights[mid] * (right - left - 1));
}
}
st.push(i);
}
}
return result;
}
};