本文主要是介绍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);}
这篇关于hihocoder 1403 后缀数组一·重复旋律 (后缀数组 + 二分)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!