上海市计算机学会竞赛平台2023年9月月赛丙组直方图

题目描述

直方图由 𝑛n 列小方格合并而成。每列有若干方格，其中第 𝑖i 列方格的数量为 ℎ𝑖hi​。请在方格图的轮廓中找到一个最大面积的矩形，输出这个矩形的面积与最大矩形的数量。

例如在上图中 𝑛=5n=5，ℎ=(3,2,2,1,2)h=(3,2,2,1,2)，最大面积为 3×2=63×2=6，只有 11 个。

输入格式

• 第一行：单个整数 𝑛n
• 第二行：𝑛n 个整数 ℎ1,ℎ2,…,ℎ𝑛h1​,h2​,…,hn​

输出格式

• 第一行：最大矩形的面积
• 第二行：最大矩形的数量

数据范围

• 30%30% 的数据，1≤𝑛≤1001≤n≤100
• 60%60% 的数据，1≤𝑛≤20,0001≤n≤20,000
• 100%100% 的数据，1≤𝑛≤300,0001≤n≤300,000
• 0≤ℎ𝑖≤500,0000≤hi​≤500,000

样例数据

输入:

5
3 2 2 1 2

输出:

6
1

详见代码:

#include <bits/stdc++.h>
using namespace std;
int n;
long long h[300005];
long long ans=0;
int cnt=0;
struct node
{
    int w;
    int h;
};
stack <node> s;
int main() 
{
    cin>>n;
    for (int i=1;i<=n;i++)
    {
        cin>>h[i];
    }
    node tmp;
    tmp.h=0;
    tmp.w=0;
    s.push(tmp);
    for (int i=1;i<=n+1;i++)
    {
        while (!s.empty()&&h[i]<=s.top().h)
        {
            int hh=s.top().h;
            s.pop();
            if (s.empty()) continue;
            long long area=hh*(i-s.top().w-1);
            if (area>ans)
            {
                ans=area;
                cnt=1;
            } 
            else if(area==ans)
            {
                cnt++;
            }
        }
        tmp.w=i;
        tmp.h=h[i];
        s.push(tmp);
    }
    cout<<ans<<endl<<cnt;
    return 0;
}

详见代码: