hihocoder 后缀自动机四·重复旋律7

题目

在\(DAG\)上跑一个\(dp\)就好了

设\(ans_i\)表示到了\(SAM\)的\(i\)位置上所有的子串形成的数的和,之后我们顺便记录一个方案数\(d_i\)

之后我们直接转移就好了

\[ans_v+=ans_u\times 10+w[u,v]\times d_u
\]

\[d_v+=d_u
\]

答案是\(\sum_{i=1}^{cnt} ans_i\)

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define maxn 1000005
#define re register
#define LL long long
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
const LL mod=1e9+7;
int n,m,cnt=1,lst=1;
char S[maxn];
int fa[maxn<<1],len[maxn<<1],son[maxn<<1][11];
LL ans[maxn<<1],d[maxn<<1];
int q[maxn<<1],top,r[maxn<<1];
inline void ins(int c)
{
int f=lst,p=++cnt; lst=p;
len[p]=len[f]+1;
while(f&&!son[f][c]) son[f][c]=p,f=fa[f];
if(!f) {fa[p]=1;return;}
int x=son[f][c];
if(len[f]+1==len[x]) {fa[p]=x;return;}
int y=++cnt;
len[y]=len[f]+1,fa[y]=fa[x],fa[x]=fa[p]=y;
for(re int i=0;i<=10;i++) son[y][i]=son[x][i];
while(f&&son[f][c]==x) son[f][c]=y,f=fa[f];
}
int main()
{
scanf("%d",&m);
for(re int i=1;i<=m;i++)
{
scanf("%s",S+1);n=strlen(S+1);
for(re int j=1;j<=n;j++) ins(S[j]-'0');
ins(10);
}
for(re int i=1;i<=cnt;i++)
for(re int j=0;j<=10;j++) r[son[i][j]]++;
q[++top]=1,d[1]=1;
for(re int i=1;i<=top;i++)
{
for(re int j=0;j<10;j++)
{
if(!son[q[i]][j]) continue;
r[son[q[i]][j]]--;
d[son[q[i]][j]]=(d[q[i]]+d[son[q[i]][j]])%mod;
ans[son[q[i]][j]]=(ans[son[q[i]][j]]+d[q[i]]*(LL)j%mod+10ll*ans[q[i]]%mod)%mod;
if(!r[son[q[i]][j]]) q[++top]=son[q[i]][j];
}
if(!son[q[i]][10]) continue;
r[son[q[i]][10]]--;
if(!r[son[q[i]][10]]) q[++top]=son[q[i]][10];
}
LL now=0;
for(re int i=2;i<=cnt;i++) now=(now+ans[i])%mod;
printf("%lld\n",now);
return 0;
}
上一篇:perl 学习笔记


下一篇:hihoCoder_1449_后缀自动机三·重复旋律6