poj1743Musical Theme - 后缀数组

题目链接:http://poj.org/problem?id=1743

题目大意:有N(1<=N<=20000)个音符的序列来表示一首乐曲,每个音符都是1..88范围内的整数,现在要找一个重复的子串,它需要满足如下条件:1.长度至少为5个音符。

2.在乐曲中重复出现(就是出现过至少两次)。(可能经过转调,“转调”的意思是主题序列中每个音符都被加上或减去了同一个整数值) 

3.重复出现的同一主题不能有公共部分。


题解:这道题是罗大神论文里的一道题,求不重复的最长子串。跑完倍增之后,二分答案,用h[]分组来验证。嗯是基本应用之一。可能有个难点就是对“转调”的处理,其实很容易可以发现同加同减的话差值还是不变,所以把原串变为两两间的差值再做sa

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define maxn 20100

int sa[maxn],rk[maxn],s[maxn],n;
int Rsort[maxn],y[maxn],wr[maxn],h[maxn];
int mymin(int x,int y){return (x<y)?x:y;}
int mymax(int x,int y){return (x>y)?x:y;}
bool cmp(int x,int y,int ln){return (wr[x]==wr[y])&&(wr[x+ln]==wr[y+ln]);}
void get_h()
{
	int i,k=0;
	for (i=1;i<=n;i++)
	{
		int j=sa[rk[i]-1];
		if (k>0) k--;
		while (s[i+k]==s[j+k]) k++;
		h[rk[i]]=k;
	}
}
void get_sa()
{
	int i,k,p,ln,m=200;
	for (i=1;i<=n;i++) rk[i]=s[i];
	for (i=0;i<=m;i++) Rsort[i]=0;
	for (i=1;i<=n;i++) Rsort[rk[i]]++;
	for (i=1;i<=m;i++) Rsort[i]+=Rsort[i-1];
	for (i=n;i>=1;i--) sa[Rsort[rk[i]]--]=i;
	ln=1;p=0;
	while (p<n)
	{
		for (k=0,i=n-ln+1;i<=n;i++) y[++k]=i;
		for (i=1;i<=n;i++) if (sa[i]-ln>0) y[++k]=sa[i]-ln;
		for (i=1;i<=n;i++) wr[i]=rk[y[i]];
		for (i=0;i<=m;i++) Rsort[i]=0;
		for (i=1;i<=n;i++) Rsort[wr[i]]++;
		for (i=1;i<=m;i++) Rsort[i]+=Rsort[i-1];
		for (i=n;i>=1;i--) sa[Rsort[wr[i]]--]=y[i];
		memcpy(wr,rk,sizeof(wr));
		p=1;rk[sa[1]]=1;
		for (i=2;i<=n;i++)
		{
			if (!cmp(sa[i],sa[i-1],ln)) p++;
			rk[sa[i]]=p;
		}m=p;ln*=2;
	}sa[0]=s[0]=0;
}
int main()
{
	//freopen("a.in","r",stdin);
	//freopen("a.out","w",stdout);
	int i,L,R,mid,mn,mx,ret;
	while (1)
	{
		scanf("%d",&n);
		if (n==0) break;
		for (i=1;i<=n;i++) scanf("%d",&s[i]);
		for (i=n;i>=1;i--) s[i]=(s[i]-s[i-1])+88;
		get_sa();get_h();//求sa[]、h[]
		L=1;R=n;ret=0;
		while (L<R)//二分
		{
			mid=(L+R)>>1;
			bool bk=0;mn=n+1;mx=0;
			for (i=1;i<=n;i++)
			 if (h[i]<mid) mx=mn=sa[i];
			 else
			 {
				 mx=mymax(mx,sa[i]);
				 mn=mymin(mn,sa[i]);
				 if (mx-mn>mid) {bk=1;break;}//要求不重叠
			 }
			if (bk) {L=mid+1;ret=mymax(ret,mid);}
			else R=mid;
		}ret++;//因为是用差值做的所以长度要+1
		if (ret<5) ret=0;
		printf("%d\n",ret);
	}
	return 0;
}


posted @ 2016-07-19 11:23  OxQ  阅读(131)  评论(0编辑  收藏  举报