描述
小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;
}