JZOJ 3648. 【GDOI2014】beyond

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]\geq 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;
}