比赛 NOIP模拟赛by mzx Day2 评测结果 AAAAAAAAAA
题目名称 拯救紫萱学姐 最终得分 100
用户昵称 liu_runda 运行时间 1.516 s
代码语言 C++ 内存使用 49.19 MiB
提交时间 2016-10-20 19:09:06
显示代码纯文本
#include<cstdio>
#include<cstring>
typedef long long ll;
const int maxn=1000005;
char str[maxn];
int next[maxn];
void getnext(){
    int j=0;
    next[0]=next[1]=0;
    for(int i=1;str[i]!='\0';++i){
        while(j!=0&&str[i]!=str[j])j=next[j];
        if(str[i]==str[j])j++;
        next[i+1]=j;
    }
}
struct edge{
    int to,next;ll w;
}lst[maxn<<1];int len=1;
int first[maxn];
void addedge(int a,int b,ll w){//printf("%d %d %lld\n",a,b,w);
    lst[len].to=b;
    lst[len].next=first[a];
    lst[len].w=w;
    first[a]=len++;
}
int T;
int q[maxn],head,tail;
int vis[maxn];ll dis[maxn];
int n;
int bfs1(){
    vis[0]=++T;
    head=tail=0;
    q[tail++]=0;
    dis[0]=0;
    while(head!=tail){
        int x=q[head++];
        for(int pt=first[x];pt;pt=lst[pt].next){
            if(vis[lst[pt].to]==T)continue;
            vis[lst[pt].to]=T;
            q[tail++]=lst[pt].to;
            dis[lst[pt].to]=dis[x]+lst[pt].w;
        }
        
    }
    int ans=0;
    for(int i=1;i<=n;++i){
        if(dis[i]>dis[ans])ans=i;
    }
    return ans;
}
ll bfs2(int s){
    vis[s]=++T;
    head=tail=0;
    q[tail++]=s;
    dis[s]=0;
    while(head!=tail){
        int x=q[head++];
        for(int pt=first[x];pt;pt=lst[pt].next){
            if(vis[lst[pt].to]==T)continue;
            vis[lst[pt].to]=T;
            q[tail++]=lst[pt].to;
            dis[lst[pt].to]=dis[x]+lst[pt].w;
        }
        
    }
    ll ans=0;
    for(int i=0;i<=n;++i){
        if(dis[i]>ans)ans=dis[i];
    }
    return ans;
}

int main(){
    freopen("savemzx.in","r",stdin);
    freopen("savemzx.out","w",stdout);
    scanf("%s",str);n=strlen(str);
    getnext();
    for(int i=1;i<=n;++i){
        addedge(next[i],i,(i-next[i])*1LL*(i-next[i]));addedge(i,next[i],(i-next[i])*1LL*(i-next[i]));
    }
    int u=bfs1();
    ll ans=bfs2(u);
    printf("%lld\n",ans);
    fclose(stdin);fclose(stdout);
    return 0;
}