POJ3261 Milk Patterns【二分】【后缀数组】

题意

给你n个数字 找至少出现t次的最长可重复子串
保证一定有答案

思路

继续刷后缀数组模板题
先二分答案，然后将后缀分成若干组。判断的是有没有一个组的后缀个数不小于t
height数组真好

AC_code:

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;
#define maxn 20005
int ss[maxn];
int sa[maxn],c[maxn];
int rank[maxn],height[maxn];
int t1[maxn], t2[maxn];
void build_sa(int s[],int n,int m) {// 字符串  字符串长度+1   最大字符+1
	int i,j,p,*x=t1,*y=t2;
	for(i=0; i<m; i++)c[i]=0;
	for(i=0; i<n; i++)c[x[i]=s[i]]++;
	for(i=1; i<m; i++)c[i]+=c[i-1];
	for(i=n-1; i>=0; i--)sa[--c[x[i]]]=i;
	for(j=1; j<=n; j<<=1) {
		p=0;
		for(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++)c[i]=0;
		for(i=0; i<n; i++)c[x[y[i]]]++;
		for(i=1; i<m; i++)c[i]+=c[i-1];
		for(i=n-1; i>=0; i--)sa[--c[x[y[i]]]]=y[i];
		swap(x,y);
		p=1;
		x[sa[0]]=0;
		for(i=1; i<n; i++)
			x[sa[i]]=y[sa[i-1]]==y[sa[i]] && y[sa[i-1]+j]==y[sa[i]+j]?p-1:p++;
		if(p>=n)break;
		m=p;
	}
}
void getHeight(int s[],int n) {
	int i,j,k=0;
	for(i=0; i<=n; i++)rank[sa[i]]=i;
	for(i=0; i<n; i++) {
		if(k)k--;
		j=sa[rank[i]-1];
		while(s[i+k]==s[j+k])k++;
		height[rank[i]]=k;
	}
}
bool check(int k, int n, int t){//长度为k  至少出现了t次的 
	int ans = 1;
	for(int i = 2; i <= n; i++){
		if(height[i] >= k){
			ans++;
			if(ans >= t){
				return true;
			}
		}else {
			ans = 1;
		}
	}
	return false;
}

int main() {
	int n, t;
	while(~scanf("%d%d", &n, &t)) {
		int maxx = -1;
		for(int i = 0; i < n; i++) {
			scanf("%d", &ss[i]);
			maxx = max(ss[i], maxx);
		}
		build_sa(ss, n+1, maxx+1);
		getHeight(ss, n);
		int ans = -1;
		int l = 1, r = n;
		while(l <= r) {
			int mid = (l+r) / 2;
			if(check(mid, n, t)) {
				ans = mid;
				l = mid + 1;
			} else {
				r = mid - 1;
			}
		}
		printf("%d\n", ans);
	}
	return 0;
}