Codeforces 547B - Mike and Feet (单调栈)

res[x]表示长度为x区间的答案,那么如果i>j,res[i]<=res[j],因此可以用长度大的区间答案去更新长度小的,所以对于每个元素,只要找到他能覆盖的最大区间长度,然后只更新这个长度的答案即可,最后再整体用长度大的更新小的一次。


对于每个元素需要找到左边和右边第一个大于这个元素的位置。单调栈可以在O(N)时间内找出所有元素的左右位置,思路就是维护一个数字单调递增的位置序列。具体看代码。


代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define LL long long
#include <stack>
using namespace std;

int N;
int a[200005];
int l[200005];
int r[200005];
int res[200005];
int main(){
    scanf("%d",&N);
    for(int i=1;i<=N;i++){
        scanf("%d",&a[i]);
    }
    stack <int> S;
    while(!S.empty()) S.pop();
    for(int i=1;i<=N;i++){
        while(!S.empty()&&a[S.top()]>=a[i]) S.pop();
        if(S.empty()) l[i]=1;
        else l[i]=S.top()+1;
        S.push(i);
    }
    while(!S.empty()) S.pop();
    for(int i=N;i>=1;i--){
        while(!S.empty()&&a[S.top()]>=a[i]) S.pop();
        if(S.empty()) r[i]=N;
        else r[i]=S.top()-1;
        S.push(i);
    }
    for(int i=1;i<=N;i++){
        int len=r[i]-l[i]+1;
        res[len]=max(res[len],a[i]);
    }
    for(int i=N-1;i>=1;i--){
        res[i]=max(res[i],res[i+1]);
    }
    for(int i=1;i<=N;i++){
        printf("%d",res[i]);
        if(i==N) printf("\n");
        else printf(" ");
    }
    return 0;
}


你可能感兴趣的:(单调栈)