hihocoder #1415 : 后缀数组三·重复旋律3 （ 后缀数组模板）

题目：小Hi平时的一大兴趣爱好就是演奏钢琴。我们知道一个音乐旋律被表示为长度为 N 的数构成的数列。小Hi在练习过很多曲子以后发现很多作品中的旋律有共同的部分。

旋律是一段连续的数列，如果同一段旋律在作品A和作品B中同时出现过，这段旋律就是A和B共同的部分，比如在abab 在 bababab 和 cabacababc 中都出现过。小Hi想知道两部作品的共同旋律最长是多少？（字符串长度不超过 100000）

后缀数组模板题

//#include<bits/stdc++.h>
#include<iostream>  
#include<cstdio>  
#include<cstring>  
#include<algorithm> 
using namespace std;
const int maxn=200010;
int t1[maxn],t2[maxn],c[maxn];
bool cmp(int *r,int a,int b,int l)
{
	return r[a]==r[b]&&r[a+l]==r[b+l];
}
void da(int str[],int sa[],int ra[],int height[],int n,int m)
{
	n++;
	int p,*x=t1,*y=t2;
	for(int i=0;i<m;i++) c[i]=0;
	for(int i=0;i<n;i++) c[x[i]=str[i]]++;
	for(int i=1;i<m;i++) c[i]+=c[i-1];
	for(int i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
	for(int j=1;j<=n;j<<=1)
	{
		p=0;
		for(int i=n-j;i<n;i++) y[p++]=i;
		for(int i=0;i<n;i++) 
		if(sa[i]>=j)  y[p++]=sa[i]-j;
		for(int i=0;i<m;i++) c[i]=0;
		for(int i=0;i<n;i++) c[x[y[i]]]++;
		for(int i=1;i<m;i++) c[i]+=c[i-1];
		for(int i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];
		swap(x,y);
		p=1;x[sa[0]]=0;
		for(int i=1;i<n;i++)
		x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
		if(p>=n) break;
		m=p; 
	}
	int k=0;
	n--;
	for(int i=0;i<=n;i++) ra[sa[i]]=i;
	for(int i=0;i<n;i++)
	{
		if(k) k--;
		int j=sa[ra[i]-1];
		while(str[i+k]==str[j+k])k++;
		height[ra[i]]=k;
	}
}

int n,k;
int ra[maxn],height[maxn],str[maxn],sa[maxn];

char s1[maxn],s2[maxn];
int main()
{
	scanf("%s%s",s1,s2);
	int n1=strlen(s1),n2=strlen(s2);
	for(int i=0;i<n1;i++)
	str[i]=s1[i]-'a'+1;
	str[n1]=27;
	for(int i=1;i<=n2;i++)
	str[n1+i]=s2[i-1]-'a'+1;
	n=n1+n2+1;
	
	da(str,sa,ra,height,n,28);
	int ans=0;
	for(int i=2;i<=n;i++)
	{
		if(sa[i]>n1&&sa[i-1]>n1) continue;
		if(sa[i]<=n1&&sa[i-1]<=n1) continue;
		ans=max(ans,height[i]);
	}
	printf("%d\n",ans);
    
/*    for(int i=1; i<=n; ++i)
		{
		      printf("%d:\t",i);
		      for(int j=i-1; j<n; ++j)
		            printf("%d",str[j]);
		      puts("");
		}
		puts("");
		puts("-------------After sort---------------");
		for(int i=1; i<=n; ++i)
		{
		      printf("sa[%2d ] = %2d\t",i,sa[i]);
		      for(int j=sa[i]-1; j<n; ++j)
		            printf("%d",str[j]);
		      puts("");
		}
		puts("");
		puts("---------------Height-----------------");
		for(int i=1; i<=n; ++i)
		      printf("height[%2d ]=%2d \n",i,height[i]);
		puts("");
		puts("----------------ra------------------");
		for(int i=1; i<=n; ++i)
		      printf("ra[%2d ] = %2d\n",i,ra[i]);
		puts("------------------END-----------------");
    */
	return 0;
}