LA 6531 Go up the Ultras 單調棧+RMQ

題意:已經懶得吐槽了。。有N個山峯,(N<=10^5),每個山峯有高度h,對應着每個山峯有一個d值,每個山峯到所有其他的嚴格比

它高的山峯都會經過一個最低值(山谷),d代表是h減去這些最低值中的最大值的差(如果不存在比它高的山峯那麼d就是它本身的

高度),問有多少山峯的d>=150000米。

思路:利用單調棧維護每個峯左邊第一個比它高的峯的位置l,右邊第一個比它高的峯的位置r,對於r,我們從前向後維護一個單調減

序列,如果當前考慮的點i比棧頂的元素高度高,那麼彈出棧頂元素,並將它的r置爲i,直到棧頂元素的高度大於等於i的高度。對於l,

從後往前維護單調減序列。對於每個點,求[ l[ i ] , i ] 的最小值d1,[ i , r[ i ] ]的最小值d2,求個最大值d=max(d1,d2),然後看是否h[i]-

d>=150000,是就輸出,可以RMQ預處理一段區間的最小值,詳見代碼:

/*********************************************************
  file name: LA6531.cpp
  author : kereo
  create time:  2015年02月06日 星期五 17時00分19秒
*********************************************************/
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<set>
#include<map>
#include<vector>
#include<stack>
#include<cmath>
#include<string>
#include<algorithm>
using namespace std;
typedef long long ll;
const int sigma_size=26;
const int N=20;
const int MAXN=100000+50;
const int inf=0x3fffffff;
const double eps=1e-8;
const int mod=100000000+7;
#define L(x) (x<<1)
#define R(x) (x<<1|1)
#define PII pair<int, int>
#define mk(x,y) make_pair((x),(y))
int n;
int dp[MAXN][N];
struct node{
    int l,r,h;
}p[MAXN];
stack<int>s;
int RMQ(int l,int r){
    int k=0;
    while((1<<(k+1))<=r-l+1) k++;
    return min(dp[l][k],dp[r-(1<<k)+1][k]);
}
int main(){
    //freopen("in.txt","r",stdin);
    while(~scanf("%d",&n)){
        for(int i=1;i<=n;i++){
            int h;
            scanf("%d",&h);
            p[i].h=dp[i][0]=h; p[i].l=p[i].r=i;
        }
        for(int j=1;(1<<j)<=n;j++){
            for(int i=1;i+(1<<j)-1<=n;i++)
                dp[i][j]=min(dp[i][j-1],dp[i+(1<<(j-1))][j-1]);
        }
        while(!s.empty())
            s.pop();
        for(int i=1;i<=n;i++){
            if(s.empty())
                s.push(i);
            else if(p[s.top()].h>=p[i].h){
                s.push(i);
            }
            else{
                while(!s.empty() && p[s.top()].h<p[i].h){
                    p[s.top()].r=i;
                    s.pop();
                }
                s.push(i);
            }
        }
        while(!s.empty())
            s.pop();
        for(int i=n;i>=1;i--){
            if(s.empty())
                s.push(i);
            else if(p[s.top()].h>=p[i].h)
                s.push(i);
            else{
                while(!s.empty() && p[s.top()].h<p[i].h){
                    p[s.top()].l=i;
                    s.pop();
                }
                s.push(i);
            }
        }
        int flag=1;
        for(int i=1;i<=n;i++){
            //int d=max(RMQ(p[i].l,i),RMQ(i,p[i].r));
            int d1=0,d2=0;
            if(p[i].l!=i)
                d1=RMQ(p[i].l,i);
            if(p[i].r!=i)
                d2=RMQ(i,p[i].r);
            int d=p[i].h-max(d1,d2);
            if(d>=150000){
                if(flag){
                    printf("%d",i);
                    flag=0;
                }
                else 
                    printf(" %d",i);
            }
        }
        printf("\n");
    }
	return 0;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章