[Bzoj3879]SvT

题意

每次询问 t t 个后缀两两的最长公共前缀之和


题解

嗯,SvT(Suffix Virtual Tree)模板, [AHOI2013] [ A H O I 2013 ] 差异的加强版

既然有 t3×106 ∑ t ≤ 3 × 10 6 然后又通过差异知道这玩意可以在后缀树上树形 DP D P 求出

那么稳的就是虚树了吧,只要每次建出虚树,然后和差异一样 DP D P 就好了

只不过就是差异可以不需要建树罢了

#include<bits/stdc++.h>
#define fp(i,a,b) for(register int i=a,I=b+1;i<I;++i)
#define fd(i,a,b) for(register int i=a,I=b-1;i>I;--i)
#define go(u) for(register int i=fi[u],v=e[i].to;i;v=e[i=e[i].nx].to)
#define file(s) freopen(s".in","r",stdin),freopen(s".out","w",stdout)
template<class T>inline bool cmax(T&a,const T&b){return a<b?a=b,1:0;}
template<class T>inline bool cmin(T&a,const T&b){return a>b?a=b,1:0;}
using namespace std;
char ss[1<<17],*A=ss,*B=ss;
inline char gc(){return A==B&&(B=(A=ss)+fread(ss,1,1<<17,stdin),A==B)?-1:*A++;}
template<class T>inline void sd(T&x){
    char c;T y=1;while(c=gc(),(c<48||57<c)&&c!=-1)if(c==45)y=-1;x=c-48;
    while(c=gc(),47<c&&c<58)x=x*10+c-48;x*=y;
}
inline void In(char*s){char c;while(c=gc(),c<32);*s++=c;while(c=gc(),c>32)*s++=c;}
char sr[1<<21],z[20];int C=-1,Z;
inline void Ot(){fwrite(sr,1,C+1,stdout),C=-1;}
template<class T>inline void we(T x){
    if(C>1<<20)Ot();if(x<0)sr[++C]=45,x=-x;
    while(z[++Z]=x%10+48,x/=10);
    while(sr[++C]=z[Z],--Z);sr[++C]='\n';
}
const int N=5e5+5,M=2*N;
typedef long long ll;
typedef int arr[M];
int n,m,ce;arr fi,pos;char s[N];
struct eg{int nx,to;}e[M];
inline void add(int u,int v){e[++ce]={fi[u],v};fi[u]=ce;}
namespace SAM{
    int las=1,T=1,ch[M][26];arr fa,len;
    inline int ins(int c){
        int p=las,np;fa[las=np=++T]=1,len[T]=len[p]+1;
        for(;p&&!ch[p][c];p=fa[p])ch[p][c]=np;
        if(p){
            int q=ch[p][c],nq;
            if(len[p]+1==len[q])fa[np]=q;
            else{
                fa[nq=++T]=fa[q],len[T]=len[p]+1;memcpy(ch[nq],ch[q],4*26);
                for(fa[q]=fa[np]=nq;ch[p][c]==q;p=fa[p])ch[p][c]=nq;
            }
        }return las;
    }
    inline void build(){fp(i,2,T)add(fa[i],i);}
}
namespace Vt{
    int k,dft;arr a,S,fa,sz,dep,dfn,son,top;ll ans;
    void dfs(int u){
        dep[u]=dep[fa[u]]+(sz[u]=1);dfn[u]=++dft;
        go(u)if(v^fa[u]){
            fa[v]=u,dfs(v),sz[u]+=sz[v];
            if(sz[v]>sz[son[u]])son[u]=v;
        }
    }
    void dfs(int u,int t){
        top[u]=t;if(son[u])dfs(son[u],t);
        go(u)if(v^fa[u]&&v^son[u])dfs(v,v);
    }
    inline int lca(int u,int v){
        for(;top[u]^top[v];dep[top[u]]>dep[top[v]]?u=fa[top[u]]:v=fa[top[v]]);
        return dep[u]<dep[v]?u:v;
    }
    inline void init(){dfs(1),dfs(1,1);memset(sz,0,sizeof sz);memset(fi,0,sizeof fi);}
    inline bool cmp(const int&a,const int&b){return dfn[a]<dfn[b];}
    void dp(int u){
        S[++S[0]]=u;
        go(u){
            dp(v),ans+=1ll*sz[u]*sz[v]*SAM::len[u];
            sz[u]+=sz[v];
        }fi[u]=0;
    }
    inline void sol(){
        sd(k);fp(i,1,k)sd(a[i]),a[i]=pos[a[i]],sz[a[i]]=1;
        sort(a+1,a+k+1,cmp);int top=1;S[1]=1;ce=S[0]=0;ans=0;
        fp(i,1,k){
            int x=a[i],p=lca(S[top],x);
            while(dep[p]<dep[S[top]]){
                if(dep[p]>=dep[S[top-1]]){
                    add(p,S[top--]);
                    if(S[top]^p)S[++top]=p;
                    break;
                }add(S[top-1],S[top]),--top;
            }if(S[top]^x)S[++top]=x;
        }while(top>1)add(S[top-1],S[top]),--top;
        dp(1);fp(i,1,S[0])sz[S[i]]=0;we(ans);
    }
}
int main(){
    #ifndef ONLINE_JUDGE
        file("s");
    #endif
    sd(n),sd(m);In(s+1);
    fd(i,n,1)
        pos[i]=SAM::ins(s[i]-'a');SAM::build();
    Vt::init();while(m--)Vt::sol();
return Ot(),0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值