jzoj3648. 【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

题解

exkmp裸题（
题目要求的是最大的i，使得a[1…i]和b[1…i]循环同构
对两个串都求出next数组，枚举i，则只有存在nextb[j]+1>=i（j=2~nexta[i]+1）时，(i-1)+(j-1)才会被计入答案
考虑倒序枚举i，同时用树状数组单调维护合法且最大的j，这样O(n log n)是可以过的
或者正序枚举i，同时用并查集每次把不合法的j删掉（与前一个合并），查找就在并查集上查找，这样是O(nα(n))的
至于判断nextb[next[a]+1]>i之类的方法，如果有证明欢迎在评论留言

code

O(n log n)

#include <algorithm>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#define fo(a,b,c) for (a=b; a<=c; a++)
#define fd(a,b,c) for (a=b; a>=c; a--)
#define low(x) (x&-(x))
using namespace std;

struct type{
	int x,s;
} b[2000001];
int S[2000001];
int T[2000001];
int f[2000001];
int g[2000001];
int F[2000001];
int G[2000001];
int tr[2000001];
int N,n,i,j,k,l,a,p,L,ans,s;
char st[4000011];
char *Ch=st;
char ch;

bool cmp(type a,type b)
{
	return a.x>b.x;
}

void swap(int &x,int &y)
{
	int z=x;
	x=y;
	y=z;
}

int Get()
{
	while (*Ch<'a' || *Ch>'z')
	*++Ch;
	
	return (*Ch++)-'a';
}

void change(int t,int s)
{
	while (t<=n)
	{
		tr[t]=max(tr[t],s);
		t+=low(t);
	}
}

int find(int t)
{
	int ans=0;
	
	while (t)
	{
		ans=max(ans,tr[t]);
		t-=low(t);
	}
	
	return ans;
}

int main()
{
//	freopen("beyond1.in","r",stdin);
//	freopen("3648.in","r",stdin);
	
	scanf("%d",&n);
	
	fread(st,1,4000011,stdin);
	fo(i,1,n)
	S[i]=Get();
	fo(i,1,n)
	T[i]=Get();
	
//	---
	
	j=1;
	while (2+j-1<=n && T[j]==T[2+j-1])
	++j;
	--j;
	g[2]=j;
	
	a=2;
	p=2+j-1;
	
	fo(i,3,n)
	if (T[1]==T[i])
	{
		L=g[i-a+1];
		
		if (i+L-1<p)
		g[i]=L;
		else
		{
			j=max(p-i+2,2);
			while (i+j-1<=n && T[j]==T[i+j-1])
			++j;
			--j;
			g[i]=j;
			
			a=i;
			p=i+j-1;
		}
	}
	
//	---
	
	j=1;
	while (1+j-1<=n && T[j]==S[1+j-1])
	++j;
	--j;
	f[1]=j;
	
	a=1;
	p=1+j-1;
	
	fo(i,2,n)
	if (T[1]==S[i])
	{
		L=g[i-a+1];
		
		if (i+L-1<p)
		f[i]=L;
		else
		{
			j=max(p-i+2,2);
			while (i+j-1<=n && T[j]==S[i+j-1])
			++j;
			--j;
			f[i]=j;
			
			a=i;
			p=i+j-1;
		}
	}
	
//	------
	
	fo(i,1,n)
	swap(S[i],T[i]);
	
	j=1;
	while (2+j-1<=n && T[j]==T[2+j-1])
	++j;
	--j;
	G[2]=j;
	
	a=2;
	p=2+j-1;
	
	fo(i,3,n)
	if (T[1]==T[i])
	{
		L=G[i-a+1];
		
		if (i+L-1<p)
		G[i]=L;
		else
		{
			j=max(p-i+2,2);
			while (i+j-1<=n && T[j]==T[i+j-1])
			++j;
			--j;
			G[i]=j;
			
			a=i;
			p=i+j-1;
		}
	}
	
//	---
	
	j=1;
	while (1+j-1<=n && T[j]==S[1+j-1])
	++j;
	--j;
	F[1]=j;
	
	a=1;
	p=1+j-1;
	
	fo(i,2,n)
	if (T[1]==S[i])
	{
		L=G[i-a+1];
		
		if (i+L-1<p)
		F[i]=L;
		else
		{
			j=max(p-i+2,2);
			while (i+j-1<=n && T[j]==S[i+j-1])
			++j;
			--j;
			F[i]=j;
			
			a=i;
			p=i+j-1;
		}
	}
	
//	------
	
	ans=f[1];
	
	fo(i,2,n)
	if (F[i])
	{
		++N;
		b[N].x=F[i];
		b[N].s=i;
	}
	sort(b+1,b+N+1,cmp);
	
	j=1;
	fd(i,n,2)
	if (f[i])
	{
		while (j<=N && b[j].x+1>=i)
		{
			change(b[j].s,b[j].s);
			++j;
		}
		
		s=find(f[i]+1);
		if (s)
		ans=max(ans,s+(i-1)-1);
	}
	
	printf("%d\n",ans);
}