Musical Theme（后缀数组 + 二分查找）

Musical Theme

音乐旋律是用N个音符组成的一个序列（1≤N≤20000）来表示，每个音符是在[1，88]范围内的整数，表示在钢琴上的一个键。然而，这样的旋律表示忽略音乐的时间概念，本题的编程工作是有关音符，与时间无关。
许多作曲家构造他们的音乐，围绕着一个重复的主题，其中，这个主题是一个完整旋律的子序列，在我们的表示中是一个整数序列。一个旋律的子序列是一个主题，如果：
至少有5个音符长
再次在乐曲的其他段出现（潜在的变调-见下文）
重复出现的主题间至少有一个是不相交的（也就是说，没有重叠在一起）
所谓变调，就是加一个正数常量或负数常量到主题子序列内的每个音符上。
给出一段旋律，计算最长主题的长度（音符的数量）。
本题解答时限为1秒。

输入

输入包含若干测试用例。每个测试案例的第一行给出整数N，接下来N个整数表示音符的序列。
最后一个测试案例后给出一个零。
使用scanf代替cin，以减少输入数据时间。

输出

对于每个测试用例，输出一行，给出一个整数，表示最长主题的长度。如果没有主题，输出0。

题解

首先，输入长度为n的字符串a，并进行预处理：相邻两个数字相减（因为题目中存在一个变调的问题），形成一个长度为n-1的新数串；
其次，计算新数串的最长公共前缀数组height；
然后，使用上述判断条件和二分查找法，计算最长重复子串的长度；
最后，若子串长度小于5，则没有主题；否则最长主题的长度为子串长度+1，因为最长重复子串是不可相邻而非不可重叠。新数串中长度为x的子串对应原数串中长度为x+1的子串，若寻找出的重复子串在新数串中相邻，则对应原串就是相交。

#include <iostream>
#include <cstring>
#include <cstdio>
#define  LL long long
#define  ULL unsigned long long
using namespace std;
const int MAXN=20100;
int wa[MAXN],wb[MAXN],wv[MAXN],Ws[MAXN];
int sa[MAXN],Rank[MAXN],height[MAXN];
int cmp(int *r,int a,int b,int l){return r[a]==r[b]&&r[a+l]==r[b+l];}
void da(int r[],int n,int m)
{
      int i,j,p,*x=wa,*y=wb,*t; 
      for(i=0; i<m; i++) Ws[i]=0;
      for(i=0; i<n; i++) Ws[x[i]=r[i]]++;
      for(i=1; i<m; i++) Ws[i]+=Ws[i-1];
      for(i=n-1; i>=0; i--) sa[--Ws[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<n; i++) wv[i]=x[y[i]];
            for(i=0; i<m; i++) Ws[i]=0;
            for(i=0; i<n; i++) Ws[wv[i]]++;
            for(i=1; i<m; i++) Ws[i]+=Ws[i-1];
            for(i=n-1; i>=0; i--) sa[--Ws[wv[i]]]=y[i];
            for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1; i<n; i++)
                  x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
      }
      return;
}
void calheight(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++);
      for(int i=n;i>=1;--i) ++sa[i],Rank[i]=Rank[i-1];
}
char s[MAXN];
int num[MAXN];
int solve(int n)
{
	int maxn=n/2,minn=0,mid,flag;
	while(minn<=maxn)
	{
		mid=(maxn+minn)/2;
		int low=sa[1],high=sa[1];
		flag=0;
		for(int i=2;i<n;i++)
			if(height[i]<mid){
				low=sa[i];
				high=sa[i];
			}else{
				low=min(low,sa[i]);
				high=max(high,sa[i]);
				if(high-low>=mid){
					flag=1;
					break;
				}
			}
		if(flag) minn=mid+1;
		else maxn=mid-1;
	}
	return maxn>=4?maxn+1:0;
}
int main()
{
	int n;
	while(scanf("%d",&n),n)
	{
		for(int i=0;i<n;i++) scanf("%d",&s[i]);
		if(n<10)
		{
			printf("0\n");
			continue;
		}
		for(int i=0;i<n-1;i++) num[i]=(s[i+1]-s[i])+90;
		int m=180;
		da(num,n,m);
		calheight(num,n-1);
		printf("%d\n",solve(n-1));
	}
	return 0;
}