BZOJ3173: [Tjoi2013]最长上升子序列 Treap 平衡树

Description

给定一个序列，初始为空。现在我们将1到N的数字插入到序列中，每次将一个数字插入到一个特定的位置。每插入一个数字，我们都想知道此时最长上升子序列长度是多少？

Input

第一行一个整数N，表示我们要将1到N插入序列中，接下是N个数字，第k个数字Xk，表示我们将k插入到位置Xk（0<=Xk<=k-1,1<=k<=N）

Output

N行，第i行表示i插入Xi位置后序列的最长上升子序列的长度是多少。

Sample Input

3
0 0 2

Sample Output

1
1
2

HINT

100%的数据 n<=100000

用Treap维护一下队列的信息，然后直接做就好。膜拜KuribohG

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
using namespace std;
const int MAXN=1000001;
struct Treap
{
    int ch[2],dat,key,size,tms;
}treap[MAXN];
int tot,root,cnt,a[MAXN],dp[MAXN],c[MAXN],ans[MAXN],n;
inline int cmp(int x,int tar)
{
    if(treap[x].dat==tar) return -1;
    return (treap[x].dat<tar?0:1);
}
inline void maintain(int x)
{
    treap[x].size=treap[treap[x].ch[0]].size+treap[treap[x].ch[1]].size+1;
}
inline int rotate(int &x,int d)
{
    int p=treap[x].ch[d^1];
    treap[x].ch[d^1]=treap[p].ch[d];
    treap[p].ch[d]=x;
    maintain(x);
    maintain(p);
    x=p;
}
void ins(int &x,int m,int tar)
{
    if(!x)
    {
        tot++;
        treap[tot].key=rand(); treap[tot].size=1; treap[tot].tms=1; treap[tot].dat=tar; x=tot;
        return;
    }
    int d;
    if(m<=treap[treap[x].ch[0]].size) ins(treap[x].ch[0],m,tar),d=0;
    else ins(treap[x].ch[1],m-treap[treap[x].ch[0]].size-1,tar),d=1;
    if(treap[treap[x].ch[d]].key>treap[x].key) rotate(x,d^1);
    maintain(x);
}
void work(int x)
{
    if(treap[x].ch[0]) work(treap[x].ch[0]);
    a[++n]=treap[x].dat;
    if(treap[x].ch[1]) work(treap[x].ch[1]);
}
int main(int argc, char *argv[])
{
    int i,x;
    scanf("%d",&n);
    for(i=1,x;i<=n;i++) {
        scanf("%d",&x);
        ins(root,x,i);
    }
    n=0;
    work(root);
    for(i=1;i<=n;i++)
    c[i]=n+1;
    for(i=1;i<=n;i++)
    {
        int sub=lower_bound(c,c+n+1,a[i])-c;
        dp[i]=sub;
        c[dp[i]]=min(c[dp[i]],a[i]);
        ans[a[i]]=dp[i];
    }
    for(i=1;i<=n;i++) ans[i]=max(ans[i],ans[i-1]);
    for(i=1;i<=n;i++) printf("%d
",ans[i]);
    return 0;
}