[JZOJ5129][SDOI省队集训2017]字符串

题目大意

从n个字符串中依次选出一个子串拼接，求本质不同的数量。

SAM

我们尽量长化前面的。
于是倒着做，对每个串都要建SAM。
只需要处理开头是某个字符的方案数，SAM的一个节点没有该转移边即可。
大概就这样，懒得写清楚了。

#include
#include
#include
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fd(i,a,b) for(i=a;i>=b;i--)
using namespace std;
const int maxn=1000000+10,mo=1000000007;
int g[maxn*2][26],step[maxn*2],fail[maxn*2],size[maxn*2],a[maxn],b[maxn];
char s[maxn],h[maxn];
int L[maxn],R[maxn];
int f[26];
int i,j,k,l,t,n,m,last,tot,top,ans,mx;
void add(int x){
    int p=last,np=++tot;
    step[np]=step[p]+1;
    while (p&&g[p][x]==0){
        g[p][x]=np;
        p=fail[p];
    }
    if (!p) fail[np]=1;
    else{
        int q=g[p][x];
        if (step[q]==step[p]+1) fail[np]=q;
        else{
            int i,nq=++tot;
            fo(i,0,25) g[nq][i]=g[q][i];
            fail[nq]=fail[q];
            fail[q]=nq;
            fail[np]=nq;
            step[nq]=step[p]+1;
            while (p&&g[p][x]==q){
                g[p][x]=nq;
                p=fail[p];
            }
        }
    }
    last=np;
}
void work(){
    int i;
    mx=0;
    fo(i,1,tot) mx=max(mx,step[i]);
    fo(i,0,mx) b[i]=0;
    fo(i,1,tot) b[step[i]]++;
    fo(i,1,mx) b[i]+=b[i-1];
    fd(i,tot,1) a[b[step[i]]--]=i;
}
void clear(){
    int i,j;
    fo(i,1,tot)
        fo(j,0,25)
            g[i][j]=0;
    fo(i,1,tot) size[i]=0;
    tot=0;
}
int main(){
    freopen("str.in","r",stdin);freopen("str.out","w",stdout);
    scanf("%d",&n);
    top=1;
    fo(i,1,n){
        L[i]=top;
        scanf("%s",h+1);
        m=strlen(h+1);
        fo(j,1,m) s[top++]=h[j];
        R[i]=top-1;
    }
    fd(i,n,1){
        last=tot=1;
        fo(j,L[i],R[i]) add(s[j]-'a');
        work();
        /*fd(j,tot,2){
            fo(k,0,25)
                if (!g[a[j]][k]){
                    size[a[j]]+=f[k];
                    if (size[a[j]]>=mo) size[a[j]]-=mo;
                }
            size[fail[a[j]]]+=size[a[j]];
            if (size[fail[a[j]]]>=mo) size[fail[a[j]]]-=mo;
        }*/
        fd(j,tot,2){
            size[a[j]]=1;
            fo(k,0,25)
                if (!g[a[j]][k]){
                    size[a[j]]+=f[k];
                    if (size[a[j]]>=mo) size[a[j]]-=mo;
                }
                else{
                    size[a[j]]+=size[g[a[j]][k]];
                    if (size[a[j]]>=mo) size[a[j]]-=mo;
                }
        }
        fo(j,0,25)
            if (g[1][j]) f[j]=size[g[1][j]];
        clear();
    }
    fo(i,0,25) (ans+=f[i])%=mo;
    ans++;
    ans%=mo;
    printf("%d\n",ans);
}