比赛 |
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;
}