美团点评2017秋招笔试编程题第3题
时间限制:1秒
空间限制:32768K
给定一组非负整数组成的数组h,代表一组柱状图的高度,其中每个柱子的宽度都为1。 在这组柱状图中找到能组成的最大矩形的面积。 入参h为一个整型数组,代表每个柱子的高度,返回面积的值。
输入描述:
输入包括两行,第一行包含一个整数n(1 ≤ n ≤ 10000) 第二行包括n个整数,表示h数组中的每个值,h_i(1 ≤ h_i ≤ 1,000,000)
输出描述:
输出一个整数,表示最大的矩阵面积。
输入例子1:
6 2 1 5 6 2 3
输出例子1:
10
分析:
枚举每一个柱子为最矮的那一个,然后找出这个柱子的左边最靠右比它矮的柱子以及右边最靠左比它矮的柱子,计算面积即可
如果往左(或往右)遇到的是比自己要大的元素,可以直接跳到这个大的元素对应的比它小的坐标数。节省了时间
#include <bits/stdc++.h> using namespace std; const int maxn=10000+5; typedef long long ll; int h[maxn],l[maxn],r[maxn]; int main() { // freopen("in.txt","r",stdin); int n; scanf("%d",&n); for(int i=1;i<=n;i++) scanf("%d",&h[i]); l[1]=0; for(int i=2;i<=n;i++) { int lp=i-1; while(h[lp]>=h[i]) lp=l[lp]; l[i]=lp; } r[n]=n+1; for(int i=n-1;i>=1;i--) { int hp=i+1; while(h[hp]>=h[i]) hp=r[hp]; r[i]=hp; } ll ans=0; for(int i=1;i<=n;i++) ans=max(ans,(ll)h[i]*(r[i]-l[i]-1)); printf("%lld\n",ans); return 0; }