Description
Input
第一行:包含一个整数N。
第二行:包含一个长度为N的字符串,字符串中只包含小写字母。
第三行:包含一个长度为N的字符串,字符串中只包含小写字母。
Output
输出答案只包含一个数字L,表示圆环最大可能有的格子数。
Sample Input
输入1:
5
abcdx
cdabz
输入2:
4
abcd
cdab
Sample Output
输出1:
4
输出2:
4
Data Constraint
对于20% 的数据,1 <= N <= 5,000
对于50% 的数据,1 <= N <= 600,000
对于100% 的数据,1 <= N <= 2,000,000
Solution
简化题意:求两个串前缀的最大循环同构子串的长度。
于是对于两个串互相各做一遍扩展KMP,求出 Extend 数组。
扫一遍,如果满足 ext1[ext2[i]+1]≥i−1 即说明满足条件(画图理解一下)。
时间复杂度 O(N) 。
Code
#include<cstdio>
using namespace std;
const int N=2e6+5;
int n,ans,mx,id;
int next[N],ext1[N],ext2[N];
char s1[N],s2[N];
inline int max(int x,int y)
{
return x>y?x:y;
}
inline void exkmp(char *s,char *t,int *ext)
{
next[1]=n;
mx=id=0;
for(int i=2;i<=n;i++)
{
int j=next[i-id+1];
if(i+j>mx)
{
j=max(mx-i+1,0);
while(i+j<=n && t[i+j]==t[j+1]) j++;
}
next[i]=j;
if(i+j-1>mx) mx=i+j-1,id=i;
}
mx=id=0;
for(int i=1;i<=n;i++)
{
int j=next[i-id+1];
if(i+j>mx)
{
j=max(mx-i+1,0);
while(i+j<=n && s[i+j]==t[j+1]) j++;
}
ext[i]=j;
if(i+j-1>mx) mx=i+j-1,id=i;
}
}
int main()
{
scanf("%d",&n);
scanf("%s",s1+1),scanf("%s",s2+1);
exkmp(s1,s2,ext1),exkmp(s2,s1,ext2);
for(int i=1;i<=n;i++)
{
int j=ext1[i];
if(ext2[j+1]>=i-1) ans=max(ans,i+j-1);
}
printf("%d",ans);
return 0;
}