比赛 NOIP模拟赛by mzx Day2 评测结果 WWAAAAWWWW
题目名称 拯救紫萱学姐 最终得分 40
用户昵称 Hzoi_Yniverse 运行时间 1.042 s
代码语言 C++ 内存使用 32.74 MiB
提交时间 2016-10-20 21:46:40
显示代码纯文本
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=1000010;
char s[maxn]; bool vis[maxn];
int next[maxn],head[maxn],n,ans,k,len;
struct Edge{ int to,next,dis; }e[maxn*2];
void Insert(int x,int y,int z){
	len++;
	e[len].to=y; e[len].dis=z;
	e[len].next=head[x]; head[x]=len;
}
void GetNext(){
	next[1]=next[0]=0;
	int j=0;
	for(int i=1;i<n;i++){
		while(j&&s[i]!=s[j]) j=next[j];
		if(s[i]==s[j]) j++;
		next[i+1]=j;
	}
}
void Build(){
	for(int i=1;i<=n;i++){
		int z=(i-next[i])*(i-next[i]);
		Insert(i,next[i],z);
		Insert(next[i],i,z);
		//printf("i==%d next==%d\n",i,next[i]);
	}
}
void Dfs(int x,int dis){
	//printf("x==%d dis==%d\n",x,dis); getchar();
	if(ans<dis){ ans=dis; k=x; }
	for(int i=head[x];i;i=e[i].next){
		int j=e[i].to;
		if(!vis[j]){//printf("x==%d j=%d\n",x,j);
			vis[j]=1; Dfs(j,dis+e[i].dis);
		}
	}
}
int main(){
	freopen("savemzx.in","r",stdin);
	freopen("savemzx.out","w",stdout);
	int __size__=128<<20;
	char *__p__=(char*)malloc(__size__)+__size__;
	__asm__("movl %0, %%esp\n"::"r"(__p__));
	scanf("%s",s); n=strlen(s);
	GetNext();
	Build();
	vis[1]=1; Dfs(1,0);
	memset(vis,0,sizeof(vis)); ans=0; //printf("k==%d\n",k);
	Dfs(k,0);
	printf("%d\n",ans);
	//while(1);
	return 0;
}