给定 n 个非负整数,用来表示柱状图中各个柱子的高度。每个柱子彼此相邻,且宽度为 1 。
求在该柱状图中,能够勾勒出来的矩形的最大面积。
以上是柱状图的示例,其中每个柱子的宽度为 1,给定的高度为 [2,1,5,6,2,3]。
图中阴影部分为所能勾勒出的最大矩形面积,其面积为 10 个单位。
示例:
输入: [2,1,5,6,2,3]
输出: 10
通过次数86,876
提交次数208,638
第一次用LeetCode交题,坑煞我也!!
放两个版本的代码,一个是LeetCode版的,一个是非LeetCode版的。
LeetCode版:
class Solution {
public:
int largestRectangleArea(vector<int>& heights) {
typedef long long ll;
const int maxn = 1e5 + 10;
int a[maxn];
int lef[maxn];
int rig[maxn];
int st[maxn];
int top = 0, n;
n = heights.size();
for(int i = 0 ;i < n; i++)
a[i + 1] = heights[i];
a[n + 1] = 0;
for(int i = n + 1; i >= 1; i--) {
while(top && a[i] <= a[st[top]]) top--;
rig[i] = st[top];
st[++top] = i;
}
top = 0;
for(int i = 1; i <= n; i++) {
while(top && a[i] <= a[st[top]]) top--;
lef[i] = st[top];
st[++top] = i;
}
ll ans = 0;
for(int i = 1; i <= n; i++) {
int l = lef[i];
int r = rig[i];
int len = r - l - 1;
ans = max(ans, 1ll * a[i] * len);
}
return ans;
}
};
非LeetCode版:
#include <iostream>
#include <cmath>
#include <cstdio>
using namespace std;
typedef long long ll;
const int maxn = 1e5 + 10;
int a[maxn];
int lef[maxn];
int rig[maxn];
int st[maxn];
int top, n;
void rig_st() {
a[n + 1] = 0;
for(int i = n + 1; i >= 1; i--) {
while(top && a[i] <= a[st[top]]) top--;
rig[i] = st[top];
st[++top] = i;
}
}
void lef_st() {
top = 0;
for(int i = 1; i <= n; i++) {
while(top && a[i] <= a[st[top]]) top--;
lef[i] = st[top];
st[++top] = i;
}
}
int main() {
cin >> n;
for(int i = 1 ;i <= n; i++)
cin >> a[i];
rig_st();
lef_st();
// for(int i = 1; i <= n; i++)
// cout << rig[i] << " ";
// cout << endl;
// for(int i = 1; i <= n; i++)
// cout << lef[i] << " ";
ll ans = 0;
for(int i = 1; i <= n; i++) {
int l = lef[i];
int r = rig[i];
int len = r - l - 1;
ans = max(ans, 1ll * a[i] * len);
}
cout << ans << endl;
return 0;
}