比赛 |
清华集训2017模板练习 |
评测结果 |
AAAAAAAAAAAAAAAAAAAA |
题目名称 |
找相同子串 |
最终得分 |
100 |
用户昵称 |
FoolMike |
运行时间 |
1.260 s |
代码语言 |
C++ |
内存使用 |
124.29 MiB |
提交时间 |
2017-07-17 16:21:40 |
显示代码纯文本
#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10;
int go[N][26],par[N],len[N],top,last;
int New(int L){len[++top]=L;return top;}
void extend(int w){
int p=last,np=New(len[p]+1);
while (p&&!go[p][w]) go[p][w]=np,p=par[p];
if (!p) par[np]=1;
else{
int q=go[p][w];
if (len[q]==len[p]+1) par[np]=q;
else{
int nq=New(len[p]+1);
memcpy(go[nq],go[q],sizeof go[q]);
par[nq]=par[q];
par[np]=par[q]=nq;
while (p&&go[p][w]==q) go[p][w]=nq,p=par[p];
}
}
last=np;
}
char s[N],t[N];
int n,m,cnt1[N],cnt2[N],cnt[N],a[N];
int main()
{
freopen("find_2016.in","r",stdin);
freopen("find_2016.out","w",stdout);
scanf("%s%s",s+1,t+1);
n=strlen(s+1);
m=strlen(t+1);
last=New(0);
for (int i=1;i<=n;i++) extend(s[i]-'a');
last=1;
for (int i=1;i<=m;i++) extend(t[i]-'a');
for (int i=1,p=1;i<=n;i++){
int w=s[i]-'a';
p=go[p][w];
cnt1[p]++;
}
for (int i=1,p=1;i<=m;i++){
int w=t[i]-'a';
p=go[p][w];
cnt2[p]++;
}
for (int i=1;i<=top;i++) cnt[len[i]]++;
for (int i=1;i<=top;i++) cnt[i]+=cnt[i-1];
for (int i=1;i<=top;i++) a[cnt[len[i]]--]=i;
long long ans=0;
for (int i=top;i;i--){
int v=a[i];
cnt1[par[v]]+=cnt1[v];
cnt2[par[v]]+=cnt2[v];
ans+=1ll*cnt1[v]*cnt2[v]*(len[v]-len[par[v]]);
}
printf("%lld\n",ans);
return 0;
}