hihocoder #1403 : 后缀数组一·重复旋律

#1403 : 后缀数组一·重复旋律

Time Limit:5000ms

Case Time Limit:1000ms

Memory Limit:256MB

描述

小Hi平时的一大兴趣爱好就是演奏钢琴。我们知道一个音乐旋律被表示为长度为 N 的数构成的数列。

小Hi在练习过很多曲子以后发现很多作品自身包含一样的旋律。旋律是一段连续的数列，相似的旋律在原数列可重叠。比如在1 2 3 2 3 2 1 中 2 3 2 出现了两次。

小Hi想知道一段旋律中出现次数至少为K次的旋律最长是多少？

解题方法提示

输入

第一行两个整数 N和K。1≤N≤20000 1≤K≤N

接下来有 N 个整数，表示每个音的数字。1≤数字≤100

输出

一行一个整数，表示答案。

Sample Input

8 2
1
2
3
2
3
2
3
1

Sample Output

4

分析：

重复两次就是height数组最大值，重复k次就是连续k-1个height数组的最小值的最大值...用单调递减的队列维护就好...

代码：

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
//by NeighThorn
using namespace std;
//guo he huan,tao sheng qian sui,hua bing di,lian kai shi zhang

const int maxn=20000+5;

int n,K,hd,ta,s[maxn],q[maxn],gs[maxn],wb[maxn],wv[maxn],sa[maxn],ran[maxn],height[maxn];

inline bool cmp(int *x,int a,int b,int l){
	return x[a]==x[b]&&x[a+l]==x[b+l];
}

inline void da(int *sa,int *x,int n,int m){
	int i,j,p,*y=wb;
	for(i=0;i<m;i++) gs[i]=0;
	for(i=0;i<n;i++) gs[x[i]]++;
	for(i=1;i<m;i++) gs[i]+=gs[i-1];
	for(i=n-1;~i;i--) sa[--gs[x[i]]]=i;
	for(j=1,p=1;p<n;j<<=1,m=p){
		for(i=n-j,p=0;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++) gs[i]=0;
		for(i=0;i<n;i++) gs[wv[i]]++;
		for(i=1;i<m;i++) gs[i]+=gs[i-1];
		for(i=n-1;~i;i--) sa[--gs[wv[i]]]=y[i];
		p=1;swap(x,y);x[sa[0]]=0;
		for(i=1;i<n;i++) x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
	}
}

inline void calheight(int n){
	int i,j,k=0;
	for(i=0;i<=n;i++) ran[sa[i]]=i;
	for(i=0;i<n;height[ran[i++]]=k)
		for(k?k--:233,j=sa[ran[i]-1];s[i+k]==s[j+k];k++);
}

inline int solve(int n){
	int hd=1,ta=0,ans=0;
	for(int i=1;i<K;i++){
		while(hd<=ta&&height[i]<height[q[ta]])
			ta--;
		q[++ta]=i;
	}
	for(int i=K;i<=n;i++){
		while(hd<=ta&&height[i]<height[q[ta]])
			ta--;
		q[++ta]=i;
		while(hd<=ta&&q[hd]<=i-K+1)
			hd++;
		ans=max(ans,height[q[hd]]);
	}
	return ans;
}

signed main(void){
	scanf("%d%d",&n,&K);
	for(int i=0;i<n;i++)
		scanf("%d",&s[i]),ran[i]=s[i];
	da(sa,ran,n+1,100),calheight(n);
	printf("%d\n",solve(n));
	return 0;
}//Cap ou pas cap. Pas cap.

---

By NeighThorn