比赛 |
清华集训2017模板练习 |
评测结果 |
AAAAAAAAAAAAAAAAAAAA |
题目名称 |
找相同子串 |
最终得分 |
100 |
用户昵称 |
SD_le |
运行时间 |
2.239 s |
代码语言 |
C++ |
内存使用 |
109.41 MiB |
提交时间 |
2017-12-02 21:39:54 |
显示代码纯文本
#include<bits/stdc++.h>
#define N 800005
using namespace std;
char ss[N],sss[N],s[N];
int l[N],tot,fa[N],ch[N][30],v1[N],v2[N];
long long ans;
int qq[N];
int n,m;
bool cmp(int x,int y)
{
return l[x]>l[y];
}
void build()
{
int lst=0;
for(int i=0;i<=n+m;i++)
{
int c=s[i]-'a';
int p=lst;int np=++tot;lst=np;
l[np]=l[p]+1;
if(i<n)v1[np]=1;
if(i>n)v2[np]=1;
while(p&&!ch[p][c])ch[p][c]=np,p=fa[p];
int q=ch[p][c];
if(!q)
{
ch[p][c]=np;
continue;
}
if(l[q]==l[p]+1)
{
fa[np]=q;
continue;
}
tot++;fa[tot]=fa[q];
fa[q]=fa[np]=tot;
l[tot]=l[p]+1;
for(int j=0;j<27;j++)ch[tot][j]=ch[q][j];
while(p&&ch[p][c]==q)ch[p][c]=tot,p=fa[p];
if(ch[p][c]==q)ch[p][c]=tot;
}
for(int i=0;i<=tot;i++)qq[i]=i;
sort(qq,qq+tot+1,cmp);
for(int i=0;i<=tot;i++)
{
v1[fa[qq[i]]]+=v1[qq[i]];
v2[fa[qq[i]]]+=v2[qq[i]];
}
for(int i=0;i<=tot;i++)
{
ans+=(long long)(l[i]-l[fa[i]])*v1[i]*v2[i];
}
return ;
}
int main()
{
freopen("find_2016.in","r",stdin);
freopen("find_2016.out","w",stdout);
scanf("%s",ss);
scanf("%s",sss);
n=strlen(ss);m=strlen(sss);
for(int i=0;i<n;i++)s[i]=ss[i];
s[n]='z'+1;
for(int i=0;i<m;i++)s[i+n+1]=sss[i];
build();
printf("%lld\n",ans);
return 0;
}