hihocoder 1403 后缀数组一·重复旋律 (后缀数组 + 二分)

时间限制:5000ms

单点时限:1000ms

内存限制: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

输出

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

样例输入

8 2
1
2
3
2
3
2
3
1

样例输出

4

题目链接：http://hihocoder.com/problemset/problem/1403

题目分析：二分答案，判断是否存在连续k-1个的height值大于等于二分值

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int const MAX = 20005;
int n, k, m, num[MAX];
int sa[MAX], rk[MAX], tp[MAX], tax[MAX], height[MAX];

bool cmp(int* r, int a, int b, int k) {
    return r[a] == r[b] && r[a + k] == r[b + k];
}

void radix_sort() {
    for (int i = 0; i <= m; i++) {
        tax[i] = 0;
    }
    for (int i = 1; i <= n; i++) {
        tax[rk[tp[i]]]++;
    }
    for (int i = 1; i <= m; i++) {
        tax[i] += tax[i - 1];
    }
    for (int i = n; i >= 1; i--) {
        sa[tax[rk[tp[i]]]--] = tp[i];
    }
}

void get_sa() {
    for (int i = 1; i <= n; i++) {
        m = max(num[i], m);
        rk[i] = num[i];
        tp[i] = i;
    }
    radix_sort();
    for (int p = 0, j = 1; p < n; j <<= 1, m = p) {
        p = 0;
        for (int i = n - j + 1; i <= n; i++) {
            tp[++p] = i;
        }
        for (int i = 1; i <= n; i++) {
            if (sa[i] > j) {
                tp[++p] = sa[i] - j;
            }
        }
        radix_sort();
        swap(rk, tp);
        rk[sa[1]] = p = 1;
        for (int i = 2; i <= n; i++) {
            rk[sa[i]] = cmp(tp, sa[i], sa[i - 1], j) ? p : ++p;
        }
    }
}

void get_height() {
    for (int i = 1, j = 0; i <= n; i++) {
        if (j) {
            j--;
        }
        int prevPos = sa[rk[i] - 1];
        while (i + j <= n && prevPos + j <= n && num[i + j] == num[prevPos + j]) {
            j++;
        }
        height[rk[i]] = j;
    }
}

bool judge(int x) {
    int count = 1;
    for (int i = 1; i <= n; i++) {
        if (height[i] >= x) {
            count++;
        } else {
            count = 1;
        }
        if (count == k) {
            return true;
        }
    }
    return false;
}

int main() {
    scanf("%d %d", &n, &k);
    for (int i = 1; i <= n; i++) {
        scanf("%d", &num[i]);
    }
    get_sa();
    get_height();
    // for (int i = 1; i <= n; i++) {
    //    printf("num[%d] = %d  sa[%d] = %d  rk[%d] = %d  height[%d] = %d\n", i, num[i], i, sa[i], i, rk[i], i, height[i]);
    // }
    int l = 0, r = n, mid = 0, ans = 0;
    while (l <= r) {
        mid = (l + r) >> 1;
        if (judge(mid)) {
            ans = mid;
            l = mid + 1;
        } else{
            r = mid - 1;
        }
    }
    printf("%d\n", ans);

}