[ 后缀自动机 树上启发式合并 线段树 树状数组 ] [ 雅礼集训 2017 Day7 ] LOJ#6041

对原串建后缀自动机, 2 2 个前缀的最长公共后缀就是其反串的后缀树中对应点的 lca len l e n 值。
现在问题转化为:给定一棵树,求编号在 [l,r] [ l , r ] 2 2 个点的 lca 的最大深度。
考虑一个暴力的做法:枚举树上每个点作为 lca l c a ,将其子树中所有点按编号排序,那么对于每个点,离它最近的 2 2 个点一定是最优的,把这些点对记下来作为三元组 (l,r,len) 表示可以用 len l e n 更新右端点 r ≥ r ,左端点 l ≤ l 的询问。求答案时双指针扫一遍,树状数组统计。
用树上启发式合并+线段树优化就好了。
时间复杂度 O(nlog2n) O ( n log 2 ⁡ n )

#include<bits/stdc++.h>
using namespace std;
inline char nc(){
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline void Read(int& x){
    char c=nc();
    for(;c<'0'||c>'9';c=nc());
    for(x=0;c>='0'&&c<='9';x=(x<<3)+(x<<1)+c-48,c=nc());
}
inline void Read(char* s){
    char c=nc();
    int len=0;
    for(;c!='0'&&c!='1';c=nc());
    for(;c=='0'||c=='1';c=nc())s[len++]=c;
}
#define fi first
#define se second
typedef pair<int,int> abcd;
const int N=200010;
int k,n,m,x,y;
int nx[N][2],f[N],d[N],lst,cnt;
int h[N],nxt[N],t[N],num;
int sz[N],son[N];
int c[N<<2];
int Ans[N],mx[N],w[N];
char s[N];
vector<abcd>g[N];
struct Node{
    int l,r,id;
}q[N];
inline void Insert(int x,int y){
    int Cur=++cnt;d[Cur]=d[lst]+1;w[Cur]=y;
    int p;
    for(p=lst;p&&!nx[p][x];p=f[p])nx[p][x]=Cur;
    if(!p)f[Cur]=1;else{
        int q=nx[p][x];
        if(d[p]+1==d[q])f[Cur]=q;else{
      int C=++cnt;d[C]=d[p]+1;
      f[C]=f[q];nx[C][0]=nx[q][0];nx[C][1]=nx[q][1];
        for(;p&&nx[p][x]==q;p=f[p])nx[p][x]=C;
        f[q]=f[Cur]=C;
    }
    }
    lst=Cur;
}
inline void Add(int x,int y){
    t[++num]=y;nxt[num]=h[x];h[x]=num;
}
void Dfs(int x){
    sz[x]=1;
    for(int i=h[x];i;i=nxt[i]){
        Dfs(t[i]);
        sz[x]+=sz[t[i]];
        if(sz[t[i]]>sz[son[x]])son[x]=t[i];
    }
}
int Find1(int x,int l,int r,int L){
    if(!c[x]||r<L)return -1;
    if(l==r)return l;
    int Mid=l+r>>1;
    if(l>=L){
        if(c[x<<1])return Find1(x<<1,l,Mid,L);
        return Find1(x<<1|1,Mid+1,r,L);
    }
    int t=Find1(x<<1,l,Mid,L);
    return ~t?t:Find1(x<<1|1,Mid+1,r,L);
}
int Find2(int x,int l,int r,int R){
    if(!c[x]||l>R)return -1;
    if(l==r)return l;
    int Mid=l+r>>1;
    if(r<=R){
        if(c[x<<1|1])return Find2(x<<1|1,Mid+1,r,R);
        return Find2(x<<1,l,Mid,R);
    }
    int t=Find2(x<<1|1,Mid+1,r,R);
    return ~t?t:Find2(x<<1,l,Mid,R);
}
void Update(int x,int l,int r,int y,int z){
    c[x]+=z;
    if(l==r)return;
    int Mid=l+r>>1;
    if(y<=Mid)Update(x<<1,l,Mid,y,z);else Update(x<<1|1,Mid+1,r,y,z);
}
inline void Modify(int x,int y){
    if(!x)return;
    int t=Find1(1,1,n,x+1);
    if(t!=-1)g[x].push_back(abcd(t,y));
    t=Find2(1,1,n,x-1);
    if(t!=-1)g[t].push_back(abcd(x,y));
    Update(1,1,n,x,1);
}
void add(int x,int y){
    Modify(w[x],y);
    for(int i=h[x];i;i=nxt[i])add(t[i],y);
}
void del(int x){
    if(w[x])Update(1,1,n,w[x],-1);
    for(int i=h[x];i;i=nxt[i])del(t[i]);
}
void Get(int x,bool fl){
    for(int i=h[x];i;i=nxt[i])
        if(t[i]!=son[x])Get(t[i],0);
    if(son[x])Get(son[x],1);
    for(int i=h[x];i;i=nxt[i])
        if(t[i]!=son[x])add(t[i],d[x]);
    Modify(w[x],d[x]);
    if(!fl)del(x);
}
inline bool Cmp(Node a,Node b){
    return a.l>b.l;
}
inline void Update(int x,int y){
    for(;x<=n;x+=x&-x)mx[x]=max(mx[x],y);
}
inline int Query(int x){
    int Ans=0;
    for(;x;x-=x&-x)Ans=max(Ans,mx[x]);
    return Ans;
}
int main(){
    Read(n);Read(m);Read(s);
    cnt=lst=1;
    for(int i=0;i<n;i++)Insert(s[i]-'0',i+1);
    for(int i=2;i<=cnt;i++)Add(f[i],i);
    Dfs(1);Get(1,1);
    for(int i=1;i<=m;i++)Read(q[i].l),Read(q[i].r),q[i].id=i;
    sort(q+1,q+m+1,Cmp);
    int j=n+1;
    for(int i=1;i<=m;i++){
        for(;j>q[i].l;){
            j--;
            for(int k=0;k<g[j].size();k++)Update(g[j][k].fi,g[j][k].se);
        }
        Ans[q[i].id]=Query(q[i].r);
    }
    for(int i=1;i<=m;i++)printf("%d\n",Ans[i]);
    return 0;
}
  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值