HDU5659：CA Loves Substring（后缀自动机）

题面
题意：一个字符集0~9的串，将第i个字符和i+1个字符分开，得到两个串。
问这两个串中有多少个不同的子串。对于每个i都要知道答案。

看到切开成两个，很自然地去考虑少了哪些串
设f[i]为切开位置为i时所少的串的个数
（大佬们说得好，正难则反）

先用SAM求出总数
少的串显然是每次出现都跨过了i的。

对于每次出现，当然就是Right集所反映的。

对于长度为len的子串x
Right集中最大的元素Maxr和最小的元素Minr

若x最前出现的区间和最后出现的区间无交
则并不会每次出现都跨过某个位置，对f无贡献

否则对两个区间相交部分的f有1的贡献
即区间$[Maxr-len+1,Minr-1]$

但我们不能枚举每个串，既然有个现成的SAM，枚举SAM的状态就可以了。
然后问题变成了区间加一个等差数列

不会区间上升线段树？
差分再差分就好了。

#include <iostream>
#include <fstream>
#include <algorithm>
#include <cmath>
#include <ctime>
#include <cstdio>
#include <cstdlib>
#include <cstring>

using namespace std;
#define mmst(a, b) memset(a, b, sizeof(a))
#define mmcp(a, b) memcpy(a, b, sizeof(b))

typedef long long LL;

const int N=400400;
const LL mo=1e9+7;

LL lj[N],f[N],g[N],ans,Violet_Evergarden;
int T,n,pre[N],son[N][10],dep[N],sr[N],br[N],last,cnt;
int to[N],nex[N],head[N],cur;
char cc[N];

void add(int u,int v)
{
    to[++cur]=v;
    nex[cur]=head[u];
    head[u]=cur;
}

void dfs(int x)
{
    for(int h=head[x];h;h=nex[h])
    {
        dfs(to[h]);
        sr[x]=min(sr[x],sr[to[h]]);
        br[x]=max(br[x],br[to[h]]);
    }
}

void Insert(int x,int op)
{
    dep[++cnt]=dep[last]+1;
    int np=cnt,p=last;
    last=cnt;
    sr[np]=br[np]=op;

    for(;!son[p][x];p=pre[p])
    son[p][x]=np;
    if(!p)
    pre[np]=1;
    else
    {
        int q=son[p][x];
        if(dep[q]==dep[p]+1)
        pre[np]=q;
        else
        {
            dep[++cnt]=dep[p]+1;
            int nq=cnt;
            sr[nq]=br[nq]=op;
            pre[nq]=pre[q];
            pre[q]=pre[np]=nq;
            mmcp(son[nq],son[q]);
            for(;son[p][x]==q;p=pre[p])
            son[p][x]=nq;
        }
    }
}

int main()
{
    lj[0]=1;
    for(int i=1;i<=200000;i++)
    lj[i]=lj[i-1]*100013ll%mo;

    cin>>T;
    while(T--)
    {

    cin>>n;

    mmst(head,0);
    mmst(f,0);
    mmst(g,0);
    Violet_Evergarden=ans=cur=0;
    last=cnt=1;
    for(int i=0;i<N;i++)
    for(int j=0;j<10;j++)
    son[i][j]=0;

    scanf("%s",cc);
    for(int i=0;i<n;i++)
    Insert(cc[i]-'0',i+1);

    for(int i=2;i<=cnt;i++)
    add(pre[i],i);
    dfs(1);

    for(int i=2;i<=cnt;i++)
    ans+=dep[i]-dep[pre[i]];

    for(int i=2;i<=cnt;i++)
    if(br[i]-dep[i]+1<=sr[i]-1)
    {
        int mi=dep[pre[i]]+1;
        if(br[i]-mi+2<=sr[i]-1)
        {
            f[br[i]-mi+2]+=dep[i]-mi+1;
            f[sr[i]]-=dep[i]-mi+1;
        }
        int hy=min(br[i]-mi+1,sr[i]-1);
        g[br[i]-dep[i]+1]++;
        g[hy+1]--;
        f[hy+1]-=hy-(br[i]-dep[i]+1)+1;
    }

    for(int i=1;i<=n-1;i++)
    {
        g[i]+=g[i-1];
        f[i]+=f[i-1];
        f[i]+=g[i];
        Violet_Evergarden=(Violet_Evergarden+(ans-f[i])%mo*lj[n-i-1]%mo)%mo;
    }
    cout<<Violet_Evergarden<<endl;

    }

    return 0;
}