POJ 1743 Musical Theme

题目大意：有N个音符（1--88的数字）表示一首歌曲，要求找两个最长的一样的主题（两段连续的数列）。

要求

两个主题长度至少是5

两个主题可以转调，意思是主题数列中每个音符都被加上或减去了同一个整数值

两个主题不可有重叠部分

思路：后缀数组，先求出相邻两个数的差值，然后就转化成了在差值数列中找两段重复的，不重叠的连续子序列

然后就是看长度的，我们可以二分答案的长度

（参考《后缀数组——处理字符串的有力工具》--罗穗骞）

#include <cstdio>
#include <iostream>
#include <cmath>
#include <algorithm>
using namespace std;
const int N = 20050;
int sa[N],rank[N],height[N],num[N];
int wa[N],wb[N],wv[N],wd[N];
int n;
bool cmp(int *r,int a,int b,int j){
	return r[a]==r[b]&&r[a+j]==r[b+j];
}
void da(int *r,int n,int m){
	int i,j,p,*x=wa,*y=wb,*t;
	for(i=0;i<m;i++)wd[i]=0;
	for(i=0;i<n;i++)wd[x[i]=r[i]]++;
	for(i=1;i<m;i++)wd[i]+=wd[i-1];
	for(i=n-1;i>=0;i--)sa[--wd[x[i]]]=i;
	for(j=1,p=1;p<n;j*=2,m=p){
		for(p=0,i=n-j;i<n;i++)y[p++]=i;
		for(i=0;i<n;i++)if(sa[i]>=j)y[p++]=sa[i]-j;
		for(i=0;i<m;i++)wd[i]=0;
		for(i=0;i<n;i++)wd[wv[i]=x[y[i]]]++;
		for(i=1;i<m;i++)wd[i]+=wd[i-1];
		for(i=n-1;i>=0;i--)sa[--wd[wv[i]]]=y[i];
		for(t=x,x=y,y=t,x[sa[0]]=0,p=i=1;i<n;i++){
			x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
		}
	}
}

void cal(int *r,int n){
	int i,j,k=0;
    for(i=1;i<=n;i++) rank[sa[i]]=i;
    for(i=0;i<n;height[rank[i++]]=k)
		for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);
}

bool solve(int len)
{	int l=N+1,r=-1;
	for(int i=1;i<n;i++){
		if(height[i]>=len){
			if(r<sa[i])r=sa[i];
			if(l>sa[i])l=sa[i];
			if(r-l>=len)return true;
		}
		else {
			l=r=sa[i];
		}
	}
	return false;
}

int main(){
	while(scanf("%d",&n)){
		if(n==0)break;
		for(int i=0;i<n;i++){
			scanf("%d",num+i);
		}
		if(n < 10) {
            printf("0\n");
            continue;
        }
        n--;
		for(int i=0;i<n;i++){
			num[i]=num[i+1]-num[i]+89;
		}
		num[n]=0;
		da(num,n+1,200);
		cal(num,n);

		int l=4,r=(n-1)/2,mid;
		while(l<r){
			mid=(l+r+1)>>1;
			if(solve(mid))l=mid;
			else r=mid-1;
		}
		int ans=l<4?0:l+1;
		printf("%d\n",ans);
	}
	return 0;
}