题目分析
后缀数组基础题目,肯定要把height数组找出来,然后从前往后扫一下,每k个需要找一个最小值,这里我用了单调队列优化,队列中一直保存着k个元素的最小元素,但是这里需要之一的一点,每一个height[i]值表示sa[i-1]与sa[i]的最长公共前缀,因此单调队列只用维护k-1个值即可。
#include <cmath>
#include <queue>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 2e4+100;
int s[maxn];
int sa[maxn],t[maxn],t2[maxn],c[maxn];
int Rank[maxn], height[maxn];
struct Node{
int id, val;
Node() {}
Node(int a, int b):id(a), val(b){}
}que[maxn];
void build_sa(int n,int m){
int i,*x = t, *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; //sa[i]中表示排名第i的位置是多少
for(int k = 1; k <= n; k <<= 1){ //k表示每次基数排序需要比较的长度,因为是按照倍增算法所以每次比较2个关键字
int p = 0;
//直接利用sa数组排序第二关键字
for(i = n-k; i < n; i++) y[p++] = i; //y中存放按第二关键字从小到大排序的位置
for(i = 0; i < n; i++) if(sa[i] >= k) y[p++] = sa[i]-k;
//基数排序第一关键字
for(i = 0; i < m; i++) c[i] = 0;
for(i = 0; i < n; i++) c[x[y[i]]]++;
for(i = 0; i < m; i++) c[i] += c[i-1];
for(i = n-1; i >= 0; i--) sa[--c[x[y[i]]]] = y[i]; //i从大到小是为了保证相同字符的情况下默认靠前的更小一些
swap(x, y); //这里只用交换指针即可
p = 1; x[sa[0]] = 0; //p表示rank值不同的字符串的数量,如果达到n表示字符串的所有关系都找出来了
for(i = 1; i < n; i++) //重新计算x的值
x[sa[i]] = y[sa[i-1]] == y[sa[i]] && y[sa[i-1]+k] == y[sa[i]+k]?p-1:p++;
if(p >= n) break;
m = p;
}
}
void getHeight(int n){
int i,j,k = 0;
for(i = 0; i < n; i++) Rank[sa[i]] = i; //求出rank值,利用rank和sa是相反的
for(i = 0; i < n; i++){
if(k) k--; //利用h[i] >= h[i-1]+1这个性质,先求出前面的后面的就可以由前面推出
j = sa[Rank[i]-1];
while(s[i+k] == s[j+k]) k++;
height[Rank[i]] = k;
}
}
int main(){
int n, k;
while(scanf("%d%d", &n, &k) != EOF){
for(int i = 0; i < n; i++) scanf("%d", &s[i]);
s[n] = 0;
build_sa(n+1, 105);
getHeight(n+1);
int head = 0, tail = -1; //维护一个单调队列
for(int i = 1; i < k; i++){
while(head <= tail && height[i] < que[tail].val) tail--;
que[++tail] = Node(i, height[i]);
}
int ans = que[head].val;
for(int i = k+1; i <= n; i++){
while(head <= tail && height[i] < que[tail].val) tail--;
que[++tail] = Node(i, height[i]);
while(head <= tail && i-que[head].id >= k-1) head++;
ans = max(ans, que[head].val);
}
printf("%d\n", ans);
}
return 0;
}