随想录Day63 | 单调栈 42. 接雨水 84.柱状图中最大的矩形

42. 接雨水

题目链接 42

给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。

第一次提交

class Solution {
public:int trap(vector<int>& height) {stack<int> mono;mono.push(-1);int res = 0;for (int i = 0; i < height.size(); i++) {if (mono.top() == -1) {mono.push(i);} else if (height[mono.top()] > height[i]) {mono.push(i);} else if (height[mono.top()] <= height[i])  {int low = height[mono.top()];while(mono.top() != -1 && height[mono.top()] <= height[i]) {int high = height[mono.top()];int width = i - mono.top()-1;res += (high - low) * width;low = high;mono.pop();} if (mono.top()!=-1){int high = height[i];int width = i - mono.top()-1;res += (high - low) * width;}mono.push(i);}cout << i << " i " << res <<" res "<<endl;}return res;}
};

学习题解

简化讨论情况

class Solution {
public:int trap(vector<int>& height) {stack<int> mono;mono.push(-1);int res = 0;for (int i = 0; i < height.size(); i++) {while(mono.top() != -1 && height[i] > height[mono.top()]) {int mid = mono.top();mono.pop();if (mono.top() != -1) {int floor = min(height[i], height[mono.top()]);int tall = floor -  height[mid];int width = i - mono.top() -1;res += tall * width;}}mono.push(i);}return res;}
};

84.柱状图中最大的矩形

题目链接 84

nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置右侧的第一个比 x 大的元素。

给你两个没有重复元素的数组 nums1 和 nums2 ，下标从 0 开始计数，其中nums1 是 nums2 的子集。

对于每个 0 <= i < nums1.length ，找出满足 nums1[i] == nums2[j] 的下标 j ，并且在 nums2 确定 nums2[j] 的下一个更大元素。如果不存在下一个更大元素，那么本次查询的答案是 -1 。

返回一个长度为 nums1.length 的数组 ans 作为答案，满足 ans[i] 是如上所述的下一个更大元素。

第一次提交

class Solution {
public:int largestRectangleArea(vector<int>& heights) {stack<int> mono;mono.push(-1);int res = 0;heights.push_back(-1);for (int i = 0; i < heights.size(); i++) {if (mono.top() == -1 || heights[i] >= heights[mono.top()]) mono.push(i);else {//cout << i << " " <<mono.top()<<endl;while(mono.top() != -1 && heights[i] < heights[mono.top()]) {int mid = heights[mono.top()];mono.pop();int width = i - mono.top() - 1;res = max(res, width* mid);//cout <<i << " "<< width << " " <<mid << endl;}mono.push(i);}}return res;}
};