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

Time Limit:15000ms

Case Time Limit:3000ms

Memory Limit:512MB

描述

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

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

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

解题方法提示

输入

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

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

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

输出

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

Sample Input

2
101
09

Sample Output

131

代码：

#include<cstdio>
#include<vector>
#include<queue>
#include<cstring>
#include<algorithm>

using namespace std;

typedef long long LL;
const LL MOD = 1000000007;
const int MAXN = 1000005; 

typedef struct Node
{
	int len, pre;
	int Next[12];
}Node;

Node tre[MAXN*2];
int cnt, last, in[MAXN*2], sum[MAXN*2];
LL ans[MAXN*2];
char str[MAXN];

void Insert(int ch)
{
	int p, q, now, rev;
	p = last, now = cnt++;
	tre[now].len = tre[last].len+1;
	while(p!=-1 && tre[p].Next[ch]==0)
	{
		tre[p].Next[ch] = now;
		p = tre[p].pre;
	}
	if(p==-1)
		tre[now].pre = 0;
	else
	{
		q = tre[p].Next[ch];
		if(tre[q].len==tre[p].len+1)
			tre[now].pre = q;
		else
		{
			rev = cnt++;
			tre[rev] = tre[q];
			tre[rev].len = tre[p].len+1;
			tre[q].pre = tre[now].pre = rev;
			while(p!=-1 && tre[p].Next[ch]==q)
			{
				tre[p].Next[ch] = rev;
				p = tre[p].pre;
			}
		}
	}
	last = now;
}

inline void Init()
{
	cnt = last = 0;
	memset(tre, 0, sizeof(tre));
	memset(in,0,sizeof in);
	memset(sum,0,sizeof sum);
	memset(ans,0,sizeof ans);
	tre[cnt++].pre = -1;
}

int main()
{
	LL all;
	int n, i, j, now, v, len;
	while(scanf("%d", &n) == 1)
	{
		Init();
		while(n--)
		{
			scanf("%s", str+1);
			for(i=1;str[i]!=0;i++)
				Insert(str[i]-'0');
			if(n!=0)
				Insert(10);
		}
		for(i=0;i<=cnt-1;i++)
		{
			for(j=0;j<=10;j++)
				in[tre[i].Next[j]]++;
		}
		queue<int> q;
		q.push(0);
		while(q.empty()==0)
		{
			now = q.front();
			q.pop();
			len = 1;
			if(now)
				len = tre[now].len-tre[tre[now].pre].len;
			for(i=0;i<=10;i++)
			{
				v = tre[now].Next[i];
				if(v!=0)
				{
					if(i!=10)
					{
						ans[v] = (ans[v]+ans[now]*10+(LL)(len-sum[now]+MOD)*i)%MOD;
						sum[v] += sum[now];
					}
					else
						sum[v] += len;
					in[v]--;
					if(in[v]==0)
						q.push(v);
				}
			}
		}
		all = 0;
		for(i=0;i<=cnt-1;i++)
			all = (all+ans[i])%MOD;
		printf("%lld\n", all);
	}
	
	return 0;
}
/*
2
12
234
out:313
*/