CodeChef - GERALD3 Chef and Substrings

题面

题意

给出一个字符串，每次询问左端点在l，r区间内的字符串一共有几种。

做法

莫队，那么只要考虑在已知道答案的L-R区间内加入一个或删除一个点的贡献即可。
难点在于去重，如果不考虑重复，那么某个点的贡献就是Len-pos+1.现在考虑重复，可以发现一个点为左端点的字符串就是以它为首的后缀的所有前缀，那么只要求出它与L-R内所有点为首的后缀的LCP的最大值即可。
这里就要用到后缀数组和高度数组，首先用st表维护一下高度数组，即可O（1）求出任意两个后缀的LCP。
将L-R内的点的后缀排名都放入一个set中去，查询时只要利用upper_bound（）查到和它最接近的两个后缀排名(一个比它大，另一个比它小），然后利用之前维护的st表求出查询点与这两个后缀的LCP的较大值，就能起到去重的效果。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<set>
#define ll long long
#define INF 0x3f3f3f3f3f3f3f3f
#define N 50010
using namespace std;

ll T,l,s,st[N],rnk[N],tmp[N],lcp[N],h,len,an[N],L,R,ans,Q,stb[N][18],lg[N];
char str[N];
struct Que
{
    ll l,r,id;
    bool operator < (const Que &u) const
    {
        return l/s<u.l/s||l/s==u.l/s&&r<u.r;
    }
}que[N];
set<int>se;
set<int>::iterator it;

inline bool cmp(ll u,ll v)
{
    if(rnk[u]!=rnk[v]) return rnk[u]<rnk[v];
    ll nu,nv;
    nu=(u+len>l)?-1:rnk[u+len];
    nv=(v+len>l)?-1:rnk[v+len];
    return nu<nv;
}

inline ll am(ll u,ll v)
{
    if(u>v) swap(u,v);
    v--;
    if(u>v) return 0;
    ll tmp=lg[v-u+1];
    return min(stb[u][tmp],stb[v-(1 << tmp)+1][tmp]);
}

inline void get()
{
    ll i,j;
    for(i=1;i<=l;i++)
    {
        stb[i][0]=lcp[i];
    }
    for(i=1;(1 << i)<=l;i++)
    {
        for(j=1;j+(1 << (i-1))<=l;j++)
        {
            stb[j][i]=min(stb[j][i-1],stb[j+(1 << (i-1))][i-1]);
        }
    }
    for(i=2,j=1;i<=50000;i++)
    {
        lg[i]=lg[i-1];
        if(i%j==0)
        {
            lg[i]++;
            j<<=1;
        }
    }
}

inline ll ask(ll u)
{
    ll res=0;
    it=se.upper_bound(rnk[u]);
    if(it!=se.end())
    {
        res=max(res,am(rnk[u],*it));
    }
    if(it!=se.begin())
    {
        it--;
        res=max(res,am(*it,rnk[u]));
    }
    return res;
}

inline void add(ll u)
{
    ll tmp=ask(u);
    ans+=l-u+1-tmp;
    se.insert(rnk[u]);
}

inline void del(ll u)
{
    se.erase(rnk[u]);
    ll tmp=ask(u);
    ans-=l-u+1-tmp;
}

int main()
{
    ll i,j;
    cin>>T;
    while(T--)
    {
        se.clear();
        scanf("%s",str+1);
        l=strlen(str+1);
        s=sqrt(l);
        for(i=1; i<=l; i++)
        {
            rnk[i]=str[i];
            st[i]=i;
        }
        for(len=1; len<=l; len<<=1)
        {
            sort(st+1,st+l+1,cmp);
            tmp[st[1]]=1;
            for(i=2; i<=l; i++)
                tmp[st[i]]=tmp[st[i-1]]+cmp(st[i-1],st[i]);
            for(i=1; i<=l; i++)
                rnk[i]=tmp[i];
        }
        lcp[0]=h=0;
        for(i=1; i<=l; i++)
        {
            j=st[rnk[i]-1];
            if(h) h--;
            for(; i+h<=l&&j+h<=l&&str[i+h]==str[j+h]; h++);
            lcp[rnk[i]-1]=h;
        }
        get();
        scanf("%lld",&Q);
        for(i=1;i<=Q;i++)
        {
            scanf("%lld%lld",&que[i].l,&que[i].r);
            que[i].id=i;
        }
        sort(que+1,que+Q+1);
        ans=0;
        for(L=i=1,R=0;i<=Q;i++)
        {
            for(;R<que[i].r;R++) add(R+1);
            for(;L>que[i].l;L--) add(L-1);
            for(;R>que[i].r;R--) del(R);
            for(;L<que[i].l;L++) del(L);
            an[que[i].id]=ans;
        }
        for(i=1;i<=Q;i++) printf("%lld\n",an[i]);
    }
}