单调栈

单调栈简单点说就是维护一个元素满足单调性的栈,即栈内元素总是单调的

 

找出序列中某一个元素 左边/右边  第一个 比它 大/小 的元素的位置

用单调栈做的话,复杂度是 O(n) 的

 

如果要求比某一元素小的 第一个 元素的位置的话,维护递减

while(!stk.empty()) stk.pop();
for(int i=0; i<N; i++){///找左边第一个比它小的、维护递增
    while(!stk.empty() && arr[stk.top()] >= arr[i]) stk.pop();
    L[i] = stk.empty() ? 0 : stk.top();///如果栈为空,说明从左边开始到这个数为止,它是最小的了
    stk.push(i);
}

while(!stk.empty()) stk.pop();
for(int i=N-1; i>=0; i--){///找右边第一个比它小的、维护递增
    while(!stk.empty() && arr[stk.top()] >= arr[i]) stk.pop();
    R[i] = stk.empty() ? N+1 : stk.top();
    stk.push(i);
}

 

如果要求比某一元素大的 第一个 元素的位置的话,维护递增

while(!stk.empty()) stk.pop();
for(int i=0; i<N; i++){///找左边第一个比它大的、维护递减
    while(!stk.empty() && arr[stk.top()] <= arr[i]) stk.pop();
    L[i] = stk.empty() ? 0 : stk.top();///如果栈为空,说明从左边开始到这个数为止,它是最大的了
    stk.push(i);
}

while(!stk.empty()) stk.pop();
for(int i=N-1; i>=0; i--){///找右边第一个比它大的、维护递减
    while(!stk.empty() && arr[stk.top()] <= arr[i]) stk.pop();
    R[i] = stk.empty() ? N+1 : stk.top();
    stk.push(i);
}

 

 

裸题在此 POJ 2559

题意 : 找出柱状图中最大的长方形的面积

分析 : 枚举每一个柱子,长方形面积的高肯定就是这个柱子的高度,宽度就是到左右延伸至小于这个柱子高度的两个柱子的位置,这就是宽了,也就是找到某一元素左右第一个比其小的元素的位置在哪里,使用单调栈就轻松解决了。

#include<bits/stdc++.h>
#define LL long long
using namespace std;
const int maxn = 1e5 + 10;

int N, L[maxn], R[maxn];
stack<int> stk;
LL H[maxn];

int main(void)
{
    while(~scanf("%d", &N) && N){
        for(int i=1; i<=N; i++) scanf("%lld", &H[i]);

        while(!stk.empty()) stk.pop();
        for(int i=1; i<=N; i++){///找左边第一个比它小的、维护递减
            while(!stk.empty() && H[stk.top()] >= H[i]) stk.pop();
            L[i] = stk.empty() ? 0 : stk.top();
            L[i]++;
            stk.push(i);
        }

        while(!stk.empty()) stk.pop();
        for(int i=N; i>=1; i--){
            while(!stk.empty() && H[stk.top()] >= H[i]) stk.pop();
            R[i] = stk.empty() ? N+1 : stk.top();
            R[i]--;
            stk.push(i);
        }

        LL ans = 0;
        for(int i=1; i<=N; i++)
            ans = max(ans, H[i] * (LL)(R[i] - L[i] + 1));

        printf("%lld\n", ans);
    }
    return 0;
}
View Code

 

posted @ 2018-04-21 16:56  qwerity  阅读(265)  评论(0编辑  收藏  举报