[Poi2010]Monotonicity 2

題目描述

給出N個正整數a[1..N]，再給出K個關係符號（>、<或=）s[1..k]。
選出一個長度爲L的子序列（不要求連續），要求這個子序列的第i項和第i+1項的的大小關係爲s[(i-1)mod K+1]。
求出L的最大值。

輸入

第一行兩個正整數，分別表示N和K (N, K <= 500,000)。
第二行給出N個正整數，第i個正整數表示a[i] (a[i] <= 10^6)。
第三行給出K個空格隔開關係符號（>、<或=），第i個表示s[i]。

輸出

一個正整數，表示L的最大值。

樣例輸入

7 3
2 4 3 1 3 5 3
< > =

樣例輸出

6

這道題顯然一看就是DP，(證明太明顯不想證)話說考試A掉這題後老師讓去講，我就去講：(恩，你們都會n^2 DP對吧，那麼怎麼優化到nlogn呢) woc？dp怎麼保證對。。

(&##！@……)呃，一臉懵逼的我，我只是來講個線段樹優化。。

呃，假裝這個DP是對的，那麼就有N^2的轉移：

for(int i=1;i<=n;i++){
    for(int j=1;j<i;j++){
        int op = f[j]%k;
        if(!op)op=k;
        if(p[op]=='<'&&a[i]>a[j])f[i]=Max(f[i],f[j]+1);
        if(p[op]=='='&&a[i]==a[j])f[i]=Max(f[i],f[j]+1);
        if(p[op]=='>'&&a[i]<a[j])f[i]=Max(f[i],f[j]+1);
    }
    Ans=Max(Ans,f[i]);
}

顯然，這是O(N^2)的算法，那麼再考慮一下如果只有<怎麼優化爲logn,當j能更新i的時候，當且僅當a[j]<a[i]且i<j 可以用樹狀數組優化，用權值樹狀數組求前綴的最大值更新就好，那麼這道題可以類似過來，可以對每一個符號建一顆線段樹，然後就可以實現O(logn)轉移了

注：等號搞個數組記錄下就好了，考試時候懶得搞了。

#include <stdio.h>
#include <cstring>
const int MAXN = 500005;
int n,k,p[MAXN],a[MAXN],Ans,f[MAXN],mx;
int tree[MAXN<<3][3];
 
 
template<typename _t>
inline _t read(){
    _t x=0,f=1;
    char ch=getchar();
    for(;ch>'9'||ch<'0';ch=getchar())if(ch=='-')f=-f;
    for(;ch>='0'&&ch<='9';ch=getchar())x=x*10+(ch^48);
    return x*f;
}
 
inline int Max(int a,int b){return a>b?a:b;}
inline int Min(int a,int b){return a<b?a:b;}
 
void Update(int o,int l,int r,int x,int val,int op){
    if(l==r){
        tree[o][op]=Max(val,tree[o][op]);
        return;
    }
    int m=l+r>>1;
    if(x<=m)Update(o<<1,l,m,x,val,op);
    else Update(o<<1|1,m+1,r,x,val,op);
    tree[o][op]=Max(tree[o<<1][op],tree[o<<1|1][op]);
}
 
int Query(int o,int l,int r,int x,int y,int op){
    if(x>y)return 0;
    if(x<=l&&r<=y)return tree[o][op];
    int m=l+r>>1,ans=0;
    if(x<=m)ans=Max(Query(o<<1,l,m,x,y,op),ans);
    if(m<y)ans=Max(Query(o<<1|1,m+1,r,x,y,op),ans);
    return ans;
}
 
int main(){
    //freopen("mot.in","r",stdin);
    //freopen("mot.out","w",stdout);
    n=read<int>();k=read<int>();
    for(int i=1;i<=n;i++)a[i]=read<int>(),mx=Max(mx,a[i]);
    for(int i=1;i<=k;i++){
        char ch=getchar();
        while(ch!='>'&&ch!='<'&&ch!='=')ch=getchar();
        switch(ch){
            case '<':p[i]=0;break;
            case '=':p[i]=2;break;
            case '>':p[i]=1;break;
        }
    }
    //T[0]  < 
	//T[1]  > 
	//T[2]  =
    Update(1,1,mx,a[1],1,p[1]);
     
    for(int i=2;i<=n;i++){
        int ans1=Query(1,1,mx,1,a[i]-1,0);
        int ans2=Query(1,1,mx,a[i]+1,mx,1);
        int ans3=Query(1,1,mx,a[i],a[i],2);
        f[i]=Max(ans1,Max(ans2,ans3))+1;
        int op = f[i]%k;
        if(!op)op=k;
        Ans=Max(f[i],Ans);
        Update(1,1,mx,a[i],f[i],p[op]);
    }
    printf("%d\n",Ans);
}