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

#1457 : 后缀自动机四·重复旋律7

时间限制:15000ms
单点时限:3000ms
内存限制:512MB

描述

小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<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
//by NeighThorn
using namespace std;

const int mod=1e9+7,maxn=2000000+5;

int n,ans,len,last=1,tail=1,in[maxn],Min[maxn],Max[maxn],cnt[maxn],sum[maxn],vis[maxn],nxt[maxn][11],fail[maxn];

char s[maxn];

inline void build(char *s){
	while(*s){
		int p=last,t=++tail,c=*s++-'0';
		Max[t]=Max[p]+1;
		while(p&&!nxt[p][c])
			nxt[p][c]=t,p=fail[p];
		if(p){
			int q=nxt[p][c];
			if(Max[q]==Max[p]+1)
				fail[t]=q,Min[t]=Max[q]+1;
			else{
				int k=++tail;
				fail[k]=fail[q];
				fail[t]=fail[q]=k;
				Max[k]=Max[p]+1;
				Min[q]=Max[k]+1;
				Min[t]=Max[k]+1;
				Min[k]=Max[fail[k]]+1;
				memcpy(nxt[k],nxt[q],11*sizeof(int));
				while(p&&nxt[p][c]==q)
					nxt[p][c]=k,p=fail[p];
			}
		}
		else
			fail[t]=Min[t]=1;
		last=t;
	}
}

inline void calc(void){
	cnt[1]=1,sum[1]=0;
	int hd=1,ta=0,q[maxn];
	q[++ta]=1;vis[1]=1;
	while(hd<=ta){
		int top=q[hd++];
		for(int i=0;i<=9;i++)
			if(nxt[top][i]){
				in[nxt[top][i]]++;
				if(!vis[nxt[top][i]])
					vis[nxt[top][i]]=1,q[++ta]=nxt[top][i];
			}
	}
	hd=1,ta=0;q[++ta]=1;
	while(hd<=ta){
		int top=q[hd++];
		for(int i=0;i<=9;i++)
			if(nxt[top][i]){
				cnt[nxt[top][i]]+=cnt[top];
				sum[nxt[top][i]]=(sum[nxt[top][i]]+sum[top]*10LL%mod+i*cnt[top])%mod;
				in[nxt[top][i]]--;
				if(!in[nxt[top][i]])
					q[++ta]=nxt[top][i];
			}
	}
	for(int i=1;i<=tail;i++)
		ans=(ans+sum[i])%mod;
	printf("%d\n",ans);
}

signed main(void){
	scanf("%d",&n);len=0;ans=0;
	for(int i=1;i<=n;i++)
		scanf("%s",s+len),len=strlen(s)+1,s[len-1]=':';
	s[len-1]='\0';
	build(s);calc();
	return 0;
}

　　

$By$  $NeighThorn$

posted @ 2017-04-20 09:05  NeighThorn  阅读(507)  评论(0)  编辑  收藏  举报