题意:给你多个串,求他们的所有子串中本质不同的串的和对2012取模
把所有串连起来建一个后缀自动机.
对于每个节点的和而言,他实际的和应加上---------过该他子节点的次数×该节点所代表的数值+他的子节点的sum×10
而实际过他这个节点的次数就更简单了,把他所有子节点的经过次数加起来就是了.
最后对所有节点求和取模即可.
这道题debug了一个多小时......最后发现是一个&&被我打成||了,emmmm
#pragma GCC optimize(2)
#pragma GCC optimize(3)
#pragma GCC optimize(4)
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 1e6 + 10;
const int M = 1e6 + 10;
struct SAM
{
static const int KN = N << 1;
static const int KM = 30;
int fail[KN], net[KN][KM], len[KN], cnt, root;
int ccnt[KN],ans[KN];
int newnode(int _len)
{
memset(net[cnt], -1, sizeof(net[cnt]));
fail[cnt] = -1;
len[cnt] = _len;
ccnt[cnt]=0;
ans[cnt]=0;
return cnt++;
}
void init()
{
cnt = 0;
root = newnode(0);
}
int add(int p, int x)
{
int np = newnode(len[p] + 1);
while(~p && net[p][x] == -1) net[p][x] = np, p = fail[p];
if(p == -1) fail[np] = root;
else
{
int q = net[p][x];
if(len[q] == len[p] + 1) fail[np] = q;
else
{
int nq = newnode(len[p] + 1);
memcpy(net[nq], net[q], sizeof(net[q]));
fail[nq] = fail[q];
fail[q] = fail[np] = nq;
while(~p && net[p][x] == q) net[p][x] = nq, p = fail[p];
}
}
return np;
}
void build(char *s, char ch)
{
int now = root;
for(int i = 0; s[i]; ++i) now = add(now, s[i] - ch);
}
int ord[KN], pri[KN];
void topo()
{
int maxVal=0;
memset(pri, 0, sizeof(pri));
for (int i = 0; i < cnt; ++i) maxVal = max(maxVal, len[i]), ++ pri[len[i]];
for (int i = 1; i <= maxVal; ++i) pri[i] += pri[i - 1];
for (int i = 0; i < cnt; ++i) ord[--pri[len[i]]] = i;
}
void gao()
{
topo();
ccnt[root]=1;
for(int i=0;i<cnt;i++)
{
int u=ord[i];
for(int j=0;j<10;j++)
{
if(~net[u][j])
{
if(i==0&&j==0) continue;
int v=net[u][j];
ccnt[v]+=ccnt[u];
ans[v]+=ans[u]*10+j*ccnt[u];
ans[v]%=2012;
}
}
}
int sum=0;
for(int i=0;i<cnt;i++) sum+=ans[i],sum=(sum+2012)%2012;
printf("%d\n",sum);
}
} sam;
char s[N];
int main()
{
int n;
while(scanf("%d",&n)!=EOF)
{
sam.init();
for(int i=0;i<n;i++) scanf("%s",s),sam.build(s, '0');
sam.gao();
}
}