Codeforces 126B Password（Z算法）

题意

给定一个字符串 (s) ，求一个子串 (t) 满足 (t) 是 (s) 的前缀、后缀且在除前缀后缀之外的地方出现过。

(1 leq |s| leq 10^6)

思路

( ext{Z})算法是一个和 ( ext{Manacher})算法很像的字符串算法，功能是求出一个 (z) 数组，代表以 (i) 开头的后缀同整个串的 ( ext{lcp}) 。

首先回顾一下 ( ext{Manacher})算法的流程。

int pos,r=0;
FOR(i,1,n)	//字符串Manacher是在原字符串每两字符间插入'#'的字符串。
{
	if(i<=r)p[i]=std::min(p[(pos<<1)-i],r-i+1);
	else p[i]=1;
	while(i-p[i]>=1&&i+p[i]<=n&&mnc[i-p[i]]==mnc[i+p[i]])p[i]++;
	if(chk_max(r,i+p[i]-1))pos=i;
}

( ext{Manacher})算法通过维护了目前扫到的最有端点，使得复杂度变成线性（可以发现把第 (3,4) 行替换成 p[i]=1; ，就变成了 (O(n^2)) 的暴力。

把它稍微变一下，就变成了( ext{Z})算法。

z[1]=n;
int l,r=0;
FOR(i,2,n)
{
	if(i<=r)z[i]=std::min(z[i-l+1],r-i+1);
	else z[i]=0;
	while(i+z[i]<=n&&str[i+z[i]]==str[1+z[i]])z[i]++;
	if(chk_max(r,i+z[i]-1))l=i;
}

仍然是通过最右端点保证复杂度，和 ( ext{Manacher})完全一个道理。

对于这道题目而言，先考虑既是前缀又是后缀的限制，只需要对原串 (str) 求一下 (z) 数组，如果一个位置 (i) 满足 (z[i]=|str|-i+1) ，那么 ([i,|str|]) 这个串就既是前缀又是后缀了。至于这个串还得在不是前缀不是后缀的地方出现过，我们可以对 (z) 数组求一个前缀最大值（除了 (z[1]) ，(z[1]) 没什么意义），这个前缀最大值就代表出现过的最长的前缀，因为你需要的串 ([i,|str|]) 肯定也是个前缀，所以就看 (|str|-i+1) 是不是小于等于前缀最大值即可。

代码

#include<bits/stdc++.h>
#define FOR(i,x,y) for(int i=(x),i##END=(y);i<=i##END;++i)
#define DOR(i,x,y) for(int i=(x),i##END=(y);i>=i##END;--i)
template<typename T,typename _T>inline bool chk_min(T &x,const _T y){return y<x?x=y,1:0;}
template<typename T,typename _T>inline bool chk_max(T &x,const _T y){return x<y?x=y,1:0;}
typedef long long ll;
const int N=1e6+5;
char str[N];int z[N];
int n;

void get_z(char *str,int n)
{
	z[1]=n;
	int l,r=0;
	FOR(i,2,n)
	{
		if(i<=r)z[i]=std::min(z[i-l+1],r-i+1);
		else z[i]=0;
		while(i+z[i]<=n&&str[i+z[i]]==str[1+z[i]])z[i]++;
		if(chk_max(r,i+z[i]-1))l=i;
	}
}

int main()
{
	scanf("%s",str+1);
	n=strlen(str+1);
	get_z(str,n);
	int r=0;
	FOR(i,2,n)
	{
		if(z[i]==n-i+1&&r>=n-i+1)
		{
			FOR(j,1,n-i+1)putchar(str[j]);
			return 0;
		}
		chk_max(r,z[i]);
	}
	puts("Just a legend");
	return 0;
}