记录编号	349356	评测结果	AAAAAAAAAA
题目名称	拯救紫萱学姐	最终得分	100
用户昵称	L_in	是否通过	通过
代码语言	C++	运行时间	0.447 s
提交时间	2016-11-14 21:23:43	内存使用	31.53 MiB

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxn 1000010
using namespace std;
typedef long long LL;
int n;
int f[maxn];
char text[maxn];
LL m1,m2;
struct Edge{
	int to,next;
	LL w;
}e[maxn<<1];
int head[maxn],tot;
void add(int from,int to,LL w){
	e[++tot].to=to;
	e[tot].w=w;
	e[tot].next=head[from];
	head[from]=tot;
}
void dfs(int x,LL dis){
	for(int i=head[x];i!=-1;i=e[i].next){
		int to=e[i].to;
		dfs(to,dis+e[i].w);
	}
	if(head[x]==-1){
		if(dis>=m1)m2=m1,m1=dis;
		if(dis<m1&&dis>m2)m2=dis;
	}
}
void get_fail(){
	for(int i=1;i<n;i++){
		int j=f[i];
		while(j&&text[i]!=text[j])j=f[j];
		f[i+1]=text[i]==text[j]?j+1:0;
	}
	for(int i=1;i<=n;i++)add(f[i],i,(f[i]-i)*1LL*(f[i]-i));
}
int main(){
	freopen("savemzx.in","r",stdin);
	freopen("savemzx.out","w",stdout);
	memset(head,-1,sizeof(head));
	scanf("%s",text);
	n=strlen(text);
	get_fail();
	dfs(0,0);
	printf("%lld\n",m1+m2);
	fclose(stdin);fclose(stdin);
	return 0;
}