[USACO5.1]乐曲主题Musical Themes

II.[USACO5.1]乐曲主题Musical Themes

一个显然的思路就是差分,这样子在原数组中差相等,就转为差分数组中子串相同

我们考虑建出后缀数组。

显然,这个答案可以二分,则我们二分一个长度\(ip\)

考虑\(ht\)数组。我们在所有\(ht_i<ip\)的地方切一刀,将\(ht\)数组切成多段,使得每段中除了第一个数,其它的\(ht_i\)全都\(\geq ip\)

显然,每段内部中任意两个串的\(\operatorname{LCP}\)的长度全都\(\geq ip\),即以它们开头\(ip\)个字符构成的字符串必定相等。

因此只要它们之间不会出现重叠,即存在一组\(|sa_i-sa_j|>ip\)时,关于\(ip\)的答案就是合法的。

因此直接二分即可。复杂度\(O(n\log n)\)

代码:

#include<bits/stdc++.h>
using namespace std;
int n,m,s[5010],x[5010],y[5010],buc[5010],sa[5010],ht[5010],rk[5010];
void SA(){
	for(int i=1;i<=n;i++)buc[x[i]=s[i]]++;
	for(int i=1;i<=m;i++)buc[i]+=buc[i-1];
	for(int i=n;i;i--)sa[buc[x[i]]--]=i;
	for(int k=1;k<=n;k<<=1){
		int num=0;
		for(int i=n-k+1;i<=n;i++)y[++num]=i;
		for(int i=1;i<=n;i++)if(sa[i]>k)y[++num]=sa[i]-k;
		for(int i=0;i<=m;i++)buc[i]=0;
		for(int i=1;i<=n;i++)buc[x[y[i]]]++;
		for(int i=1;i<=m;i++)buc[i]+=buc[i-1];
		for(int i=n;i;i--)sa[buc[x[y[i]]]--]=y[i],y[i]=0;
		swap(x,y),x[sa[1]]=num=1;
		for(int i=2;i<=n;i++)x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k])?num:++num;
		m=num;
	}
	for(int i=1;i<=n;i++)rk[sa[i]]=i;
	for(int i=1,k=0;i<=n;i++){
		if(rk[i]==1)continue;
		if(k)k--;
		int j=sa[rk[i]-1];
		while(i+k<=n&&j+k<=n&&s[i+k]==s[j+k])k++;
		ht[rk[i]]=k;
	}
}
bool che(int ip){
	for(int i=2,j=2;i<=n;i=j){
		int mn=sa[i],mx=sa[i];
		j=i+1;
		while(ht[j]>=ip)mn=min(mn,sa[j]),mx=max(mx,sa[j]),j++;
		if(mx-mn>ip)return true;
	}
	return false;
}
int main(){
	scanf("%d",&n),n--;
	for(int i=0;i<=n;i++)scanf("%d",&s[i]);
	for(int i=n;i;i--)s[i]-=s[i-1];
	for(int i=1;i<=n;i++)s[i]+=88;//add 88 to make it positive, then we can bucket sort it more easily.
	m=200;
	SA();
	int l=0,r=n;
	while(l<r){
		int mid=(l+r+1)>>1;
		if(che(mid))l=mid;
		else r=mid-1;
	}
	l++;
	printf("%d\n",l<5?0:l);
	return 0;
}

posted @ 2021-03-31 16:48  Troverld  阅读(54)  评论(0编辑  收藏  举报