[模板] LIS

树状数组优化LIS到nlogn,网上找了好多,感觉讲得都不是很明白,正好自己复习整理一下。

基本的DP方程 f[i]=max(f[i],f[j]+1) (j<i且a[j]<a[i])

定义一个数组b
以a[i]为下标
存储 以a[i]结尾的最长上升子序列长度

例如 序列 5 2 1 3 4
最终b数组如下

i 1 2 3 4 5
b 1 1 2 3 1

定义它有什么好处呢?
每次转移f[i],是找f[j]的过程,有两个条件:
1. f[j]最大值
2. j<i
第二个j<i好说,用循环卡范围在[1,i-1]即可。
那第一个条件呢?a在[1,i-1]的最大值,区间最值,RMQ问题?
用什么呢?ST表?不行,我们要更改。

注意到这个区间一定是一个前缀区间,那树状数组呢? 树状数组本质是求前缀和的,当然可以改造成求前缀最大值啦。

void query(int now){
    int ret=0;
    while(now){
        ret=max(ret,t[now]);
        now-=now&-now;
    }
    return ret;
} 

void updata(int now,int w){
    while(now<=n){
        t[now]=max(t[now],w);
        now+=now&-now;
    }
}

再具体考虑一下转移,先考虑限制a[j]<a[j]的问题

把a数组离散化,1-n个下标对应1-n个数

定义v为a[i]
那就是求出b[1],b[2],… ,b[v-1]的最大值,这个可以用树状数组。
那j<i的问题呢?
换言之,b[1~v-1]是可以求最大值,那怎么*保证这些比a[i]小的数一定在i前面*呢?

答案是:不需要保证

在i前面的数,之前已经更新过,所以b数组中这些数有了值,而i之后的数,尚未访问过,就是0。
0是不影响最大值的。

所以,我们的大体思路已经成型了:

  1. 求出b[1],b[2],…,b[v-1]的最大值

  2. 用这个最大值+1更新b[v]

再次强调上述的v是里当前正在更新的那个数,也就是a[i],i是下标,v是树状数组的下标。

所以,这个b数组不需要真实存在,用树状数组假装它存在,实际上维护前缀最大值就行。

复杂度估计一下,循环遍历n个数,每次找前缀最大值logn,合计O(nlogn)

//Stay foolish,stay hungry,stay young,stay simple
#include<iostream>
#include<algorithm> 
using namespace std;

const int MAXN=20000;

int a[MAXN],tmp[MAXN],f[MAXN];
int n;

int t[MAXN];
inline void updata(int now,int w){
    while(now<=n){
        t[now]=max(t[now],w);
        now+=now&-now;
    }
}
inline int query(int now){
    int ret=0;
    while(now){
        ret=max(ret,t[now]);
        now-=now&-now;
    }
    return ret;
}

int main(){
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        tmp[i]=a[i];
    }
    sort(tmp+1,tmp+1+n);
    int tot=unique(tmp+1,tmp+1+n)-tmp;
    for(int i=1;i<=n;i++)
        a[i]=lower_bound(tmp+1,tmp+1+tot,a[i])-tmp;
    int ans=1;
    for(int i=1;i<=n;i++){
        f[i]=query(a[i]-1)+1;
        updata(a[i],f[i]);
        ans=max(f[i],ans);
    }
    cout<<ans<<endl; 
    return 0;
}

本文来自博客园,作者:GhostCai,转载请注明原文链接:https://www.cnblogs.com/ghostcai/p/9247447.html

原文地址:https://www.cnblogs.com/ghostcai/p/9247447.html