比赛 清华集训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;
}