hdu 1506(单调栈)

利用单调栈,可以在O(n)的复杂度下算出一个数组每个元素左/右第一个比它大/小的元素的位置

#include<cstdio>
#include<iostream>
#include<stack>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N = 1e5+9;
stack <int> s;
int l[N],a[N],r[N];
ll h[N];
int main()
{
    freopen("in.txt","r",stdin);
    ios::sync_with_stdio(false);
    while(true){
       // memset(l,0,sizeof(l));
       // memset(r,0,sizeof(r));
        int n; cin>>n; if(n==0) break;
        for(int i = 1; i<=n; i++) cin>>h[i];
        while(s.size()) s.pop();
        for(int i = 1; i<=n; i++){
            while(s.size()&&h[s.top()]>=h[i]) s.pop();
            if(s.empty()) l[i] = 0;
            else  l[i] = s.top();
            s.push(i);
        }
        while(s.size()) s.pop();
        for(int i = n; i>0; i--){
            while(s.size()&&h[s.top()]>=h[i]) s.pop();
            if(s.empty()) r[i] = n+1;
            else r[i] = s.top();
            s.push(i);
        }
    //    for(int i = 1; i<=n; i++)
    //        cout<<l[i]<<' ';
    //    cout<<endl;
    //    for(int i = 1; i<=n; i++)
    //        cout<<r[i]<<' ';
    //    cout<<endl;
        ll ans = 0;
        for(int i = 1; i<=n; i++){
            ans = max(ans,h[i]*(r[i]-l[i]-1));
        }
        cout<<ans<<endl;
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_41156122/article/details/81605849