poj2082 terrible sets 单调栈

题目链接：https://vjudge.net/problem/POJ-2082

题意：给定一些矩形的底边长和高，问能拼出的最大新矩形的面积

等价于求一个区间内最小高度乘底边宽度和的最大值。考虑每个矩形，向左右延伸至第一个高度小于它的矩形，记录下延伸到的前一个位置l[i]和r[i]，用类似前缀和的方法求出几个连续矩形的底边宽度和。

则答案ans=max(ans,(s[r[i]]-s[l[i]-1])*a[i].h)，此处s即为前缀和数组。

注意要开long long

#include<iostream>
#include<algorithm>
#include<stack>
#include<cstring>
#define ll long long
using namespace std;
const int maxn=500+10;

struct qwq{int w,h;}a[maxn];
stack<int> st;
int l[maxn],r[maxn],s[maxn];
int t,n,i,j,k,m;

void solve(){
    memset(s,0,sizeof(s));
    for (i=1;i<=n;i++) s[i]=s[i-1]+a[i].w;
    st.push(1);
    a[n+1].h=a[0].h=-1;
    for (i=2;i<=n+1;i++){ //向右延伸 
        while (!st.empty()&&a[st.top()].h>a[i].h){
            r[st.top()]=i-1;
            st.pop();    
        }
        st.push(i);
    }
    while (!st.empty()) st.pop();st.push(n);
    for (i=n-1;i>=0;i--){ //向左延伸        
        while (!st.empty()&&a[st.top()].h>a[i].h){
            l[st.top()]=i+1;
            st.pop();
        }    
        st.push(i);
    }
    ll ans=0;
    for (i=1;i<=n;i++) { 
        ll res=(s[r[i]]-s[l[i]-1])*a[i].h;
        ans=max(ans,res);
    }
    cout<<ans<<endl;
    while (!st.empty()) st.pop();
}

int main(){
    //freopen("poj2082.txt","r",stdin);
    std::ios::sync_with_stdio(false);
    cin>>n;
    while (n!=-1) {
        for (i=1;i<=n;i++) cin>>a[i].w>>a[i].h;
        solve();
        cin>>n;
    }
    return 0;
}