【hihocoder1457】后缀自动机四·重复旋律7 后缀自动机

描述

小Hi平时的一大兴趣爱好就是演奏钢琴。我们知道一段音乐旋律可以被表示为一段数构成的数列。

神奇的是小Hi发现了一部名字叫《十进制进行曲大全》的作品集，顾名思义，这部作品集里有许多作品，但是所有的作品有一个共同特征：只用了十个音符，所有的音符都表示成0-9的数字。

现在小Hi想知道这部作品中所有不同的旋律的“和”（也就是把串看成数字，在十进制下的求和，允许有前导0）。答案有可能很大，我们需要对（10^9 + 7)取摸。

解题方法提示

输入

第一行，一个整数N，表示有N部作品。

接下来N行，每行包含一个由数字0-9构成的字符串S。

所有字符串长度和不超过 1000000。

输出

共一行，一个整数，表示答案 mod （10^9 + 7)。

样例输入
2
101
09
样例输出
131

代码

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#define ll long long
#define mod 1000000007
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
const int N=2000005;
int n,cnt,last,S,ans,len;
int sum[N],l[N],trans[N][11],fa[N];
char s[N],ch[N];
int t[N],q[N],du[N];
void add(int x)
{
    int p=last,np=++cnt,c=(s[x]!='!')?s[x]-'0':10;last=cnt;l[np]=x;
    for (;p&&!trans[p][c];p=fa[p]) trans[p][c]=np;
    if (!p) fa[np]=S;
    else
    {
        int q=trans[p][c];
        if (l[p]+1==l[q]) fa[np]=q;
        else
        {
            int nq=++cnt;l[nq]=l[p]+1;
            memcpy(trans[nq],trans[q],sizeof(trans[q]));
            fa[nq]=fa[q];fa[q]=fa[np]=nq;
            for (;trans[p][c]==q;p=fa[p]) trans[p][c]=nq;
        }
    }
}
void solve()
{
    for (int i=1;i<=cnt;i++) t[l[i]]++;
    for (int i=1;i<=len;i++) t[i]+=t[i-1];
    for (int i=cnt;i;i--) q[t[l[i]]--]=i;du[1]=1;
    for (int i=1;i<=cnt;i++)
    {
        int x=q[i];if (!du[x]) continue;
        for (int c=0;c<=9;c++)if (trans[x][c])
        {
            (du[trans[x][c]]+=du[x])%=mod;
            (sum[trans[x][c]]+=((ll)sum[x]*10%mod+(ll)c*du[x]%mod)%mod)%=mod;
        }
    }
}
int main()
{
    n=read();last=S=++cnt;
    for (int i=1;i<=n;i++)
    {
        scanf("%s",ch);len=strlen(s+1);
        if (len) s[len+1]='!';int l=strlen(ch);
        for (int j=0;j<l;j++) s[len+1+j+((len)?1:0)]=ch[j];
    }
    len=strlen(s+1);
    for (int i=1;i<=len;i++) add(i);
    solve();
    for (int i=1;i<=cnt;i++) (ans+=sum[i])%=mod;
    printf("%d",ans);
    return 0;
}