[BZOJ 3173] [TJOI 2013] 最长上升子序列(fhq treap)

[BZOJ 3173] [TJOI 2013] 最长上升子序列(fhq treap)

题面

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

分析

这题有几个重要性质：

第一个性质是，插入的数的大小是递增的。我们插入第i个数后的答案，为以当前序列中的数为结尾的LIS长度的最大值，而根据当前序列中的数都在[1,i]内，们可以维护以值v为结尾的LIS长度len[i],那么第i次插入后的答案就是\(max(len[j]) (j \in [1,i])\)

第二个性质是，后插入的数不会影响以前面插入的数为结尾的LIS长度。这句话有点抽象，我们来举一个例子。插入完两个数后的序列是{1,2},以2结尾的LIS长度为2。然后我们在2前面插入3，序列变成了{1,3,2},因为3>2,它不会影响以2结尾的LIS长度，因此以2结尾的LIS长度仍为2。如果插入在2后面，显然新的LIS结尾不可能是2，对以2结尾的LIS长度没有影响。

因此，我们只要用一个数据结构处理出插入n次后的最终序列，对这个序列跑一次LIS，求出以v结尾的序列长度，然后前缀max一下求出答案。这里我选择了fhq treap来维护序列，然后用模板的\(O(n \log n)\)求LIS算法。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#define maxn 100000 
using namespace std;
struct fhq_treap{
#define lson(x) tree[x].ls
#define rson(x) tree[x].rs
    struct node{
        int ls;
        int rs;
        int val;
        int sz;
        int dat;
    }tree[maxn+5];
    int root;
    int ptr;
    int New(int val){
        ptr++;
        tree[ptr].val=val;
        tree[ptr].dat=rand();
        tree[ptr].sz=1;
        return ptr;
    }
    void push_up(int x){
        tree[x].sz=tree[lson(x)].sz+tree[rson(x)].sz+1;
    } 
    int merge(int x,int y){
        if(!x||!y) return x+y;
        if(tree[x].dat<tree[y].dat){
            tree[y].ls=merge(x,tree[y].ls);
            push_up(y);
            return y;
        }else{
            tree[x].rs=merge(tree[x].rs,y);
            push_up(x);
            return x;
        } 
    }
    void split(int now,int k,int &x,int &y){
        if(now==0){
            x=y=0;
            return;
        }
        if(k<=tree[lson(now)].sz){
            y=now;
            split(tree[now].ls,k,x,tree[y].ls);
        }else{
            x=now;
            split(tree[now].rs,k-tree[lson(now)].sz-1,tree[x].rs,y); 
        }
        push_up(now);
    }
    void insert(int val,int pos){
        int x,y;
        split(root,pos-1,x,y);
        root=merge(merge(x,New(val)),y);
    }
    void print(int x,int *out,int &sz){
        if(x==0) return;
        print(lson(x),out,sz);
        out[++sz]=tree[x].val;
        print(rson(x),out,sz);
    }
}T;

int n;
int sz=0;
int a[maxn+5];

int top=0;
int s[maxn+5];
int len[maxn+5];//以值i为结尾的LIS长度 
void get_lis(){
    for(int i=1;i<=n;i++){
        if(a[i]>s[top]){
            s[++top]=a[i];
            len[a[i]]=top;
        }else{
            int tmp=lower_bound(s+1,s+1+top,a[i])-s;
            s[tmp]=a[i];
            len[a[i]]=tmp;
        } 
    }
} 
int main(){
    int x;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&x);
        x++;
        T.insert(i,x);
    }
    T.print(T.root,a,sz);
//  for(int i=1;i<=sz;i++) printf("%d ",a[i]);
    get_lis();
    int ans=0;
    for(int i=1;i<=n;i++){
        ans=max(ans,len[i]);
        printf("%d\n",ans);
        //第i个数插入的时候，序列里只有1~i的数，把以它们为结尾的lis长度取max即可 
    } 
}