[LOJ]#2083. 「NOI2016」优秀的拆分 二分+hash+思路

Solution

其实这个东西也不知道是称为思路还是套路……
就是在做这类要在字符串中找相同的两段的题目，有一个这样的做法：枚举串的长度$L$，然后把字符串分段，每$L$个分一段，把每一段的第一个位置记为关键位置，这样每个长度为$L$的子串都会经过且仅经过一个关键位置，然后对每个关键位置$p$和下一个关键位置$p+L$，求前缀的$S[1..p]、S[1..p+L]$的最长公共后缀和后缀$S[p..n]、S[p+L..n]$的最长公共前缀，然后就得到了包含$p$和$p+L$的两段相同的串$S[l_1..r_1]、S[l_2..r_2]$，当然要限制这两个串不能包含一个以上关键位置，然后就得到这两串任意一个长度为$L$的对应子串都是相同的，然后就可以做很多事情。
对于这道题目，可以对于每个位置算出以这个位置为起点和终点的形如$AA$的串有多少个，然后最后再算答案，这个可以用上面的东西加上简单的差分实现。
找$LCP、LCS$如果是二分+hash，复杂度为$O(n{\log }^{2}n)$，如果用后缀数组，复杂度为$O(n\log n)$。

Code

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define pa pair<int,int>
const int Maxn=60010;
const int inf=2147483647;
const int base=233,mod=1000000007;
int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
	return x*f;
}
int n;char str[Maxn];
int Hash[Maxn],Pow[Maxn],s1[Maxn],s2[Maxn];
int g(int l,int r){return(Hash[l]-(LL)Hash[r+1]*Pow[r-l+1]%mod+mod)%mod;}
int lcp(int x,int y)
{
	int l=1,r=n-y+1;
	while(l<=r)
	{
		int mid=l+r>>1;
		if(g(x,x+mid-1)==g(y,y+mid-1))l=mid+1;
		else r=mid-1;
	}
	return l-1;
}
int lcs(int x,int y)
{
	int l=1,r=x;
	while(l<=r)
	{
		int mid=l+r>>1;
		if(g(x-mid+1,x)==g(y-mid+1,y))l=mid+1;
		else r=mid-1;
	}
	return l-1;
} 
int main()
{
	int T=read();
	Pow[0]=1;for(int i=1;i<=30000;i++)Pow[i]=(LL)Pow[i-1]*base%mod;
	while(T--)
	{
		LL ans=0;
		scanf("%s",str+1);
		n=strlen(str+1);
		Hash[n+1]=0;for(int i=n;i;i--)Hash[i]=((LL)Hash[i+1]*base%mod+str[i]-'a'+1)%mod;
		for(int L=1;L<(n>>1);L++)
		{
			for(int x=1;x+L<=n;x+=L)
			{
				int y=x+L;
				if(str[x]!=str[y])continue;
				int l=min(lcs(x,y),L),r=min(lcp(x,y),L);
				if(l+r-1<L)continue;
				s1[x-l+1]++,s1[x+r-L+1]--;
				s2[x-l+(L<<1)]++,s2[x+r+L]--;
			}
		}
		for(int i=1;i<=n;i++)
		{
			s1[i]+=s1[i-1],s2[i]+=s2[i-1];
			ans+=(LL)s1[i]*s2[i-1];
		}
		for(int i=0;i<=(n<<1);i++)s1[i]=s2[i]=0;
		printf("%lld\n",ans);
	}
}