最少出现K次我们可以用Height数组的lcp来得出,而恰好出现K次,我们只要除去最少出现K+1次的lcp即可。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std; const int maxn = + ; int t1[maxn], t2[maxn], c[maxn]; bool cmp(int* r, int a, int b, int l) {
return r[a] == r[b] && r[a + l] == r[b + l];
} void da(int str[], int sa[], int Rank[], int lcp[], int n, int m) {
++n;
int i, j, p, *x = t1, *y = t2;
for (i = ; i < m; ++i) c[i] = ;
// puts("hha");
for (i = ; i < n; ++i) c[x[i] = str[i]]++;
for (i = ; i < m; ++i) c[i] += c[i - ];
for (i = n - ; i >= ; --i) sa[--c[x[i]]] = i;
for (j = ; j <= n; j <<= ) {
p = ;
for (i = n - j; i < n; ++i) y[p++] = i;
for (i = ; i < n; ++i) if (sa[i] >= j) y[p++] = sa[i] - j;
for (i = ; i < m; ++i) c[i] = ;
for (i = ; i < n; ++i) c[x[y[i]]]++; for (i = ; i < m; ++i) c[i] += c[i - ];
for (i = n - ; i >= ; --i) sa[--c[x[y[i]]]] = y[i]; swap(x, y);
p = ; x[sa[]] = ;
for (i = ; i < n; ++i) {
x[sa[i]] = cmp(y, sa[i - ], sa[i], j) ? p - : p++;
}
if (p >= n)break;
m = p;
} int k = ;
n--;
for (i = ; i <= n; ++i) Rank[sa[i]] = i;
for (i = ; i < n; ++i) {
if (k)--k;
j = sa[Rank[i] - ];
while (str[i + k] == str[j + k])++k;
lcp[Rank[i]] = k;
//cout << k << endl;
}
} int lcp[maxn], a[maxn], sa[maxn], Rank[maxn]; char s[maxn]; int d[maxn][];
int len; void rmq_init(int* A, int n) {
for (int i = ; i < n; ++i) d[i][] = A[i];
for (int j = ; ( << j) <= n; ++j)
for (int i = ; i + ( << j) - < n; ++i)
d[i][j] = min(d[i][j - ], d[i + ( << (j - ))][j - ]);
} int ASK(int l, int r) {
int k = ;
while (( << (k + )) <= r - l + )++k;
return min(d[l][k], d[r - ( << k) + ][k]);
} int ask(int l, int r) {
if (l == r) return len - sa[r]; /// l == r的话 是一个串, 返回本身的长度即可。
return ASK(l + , r); ///否则在rmq查询。
} //
int main() {
//freopen("in.txt", "r", stdin);
//freopen("outstd.txt", "w", stdout);
int T;
scanf("%d", &T);
while (T--) {
int k;
scanf("%d", &k);
scanf("%s", s);
//puts(s);
len = strlen(s);
for (int i = ; i < len; ++i) {
a[i] = s[i] - 'a' + ;
}
a[len] = ;
da(a, sa, Rank, lcp, len, );
rmq_init(lcp, len + );
long long ans = ;
if (k == )
{
for (int i = ; i <= len; i++)
{
int siz = len - sa[i];
int des = ;
if (i > ) des = max(des, ask(i - , i));
if (i < len) des = max(des, ask(i, i + ));
ans += siz - des;
}
}
else
{
for (int i = ; i+k- <= len; i++)
{
int siz = ask(i, i + k - );
int des = ;
if (i > ) des = max(des, ask(i - , i + k - ));
if (i + k <= len) des = max(des, ask(i, i + k));
ans += siz - des;
}
}
printf("%I64d\n", ans);
/*
long long ans = 0; for (int i = 1; i + k - 1 <= len; ++i) {
ans += ask(i, i + k - 1);
if (i - 1 > 0)ans -= ask(i - 1, i + k - 1); ///注意边界问题。
if (i + k <= len)ans -= ask(i, i + k);
if (i - 1 > 0 && i + k <= len)ans += ask(i - 1, i + k);
}
printf("%I64d\n", ans);*/ }
return ;
}

最新文章

  1. Java2_J2EE体系架构
  2. No enum constant org.apache.ibatis.type.JdbcType.xxx 错误
  3. outlook——还原“未读邮件”文件夹
  4. 将数据集做成VOC2007格式用于Faster-RCNN训练
  5. .net学习笔记---IIS 处理模型及ASP.NET页面生命周期
  6. Objective-C:Foundation框架-常用类-NSArray
  7. C#序列化/反序列化
  8. 提示框的优化之自定义Toast组件之(三)Toast组件优化
  9. Jquery 类似新浪微博,鼠标移到头像,用浮动窗口显示用户信息,已做成一个jquery插件
  10. 用友CDM系统,将货位间商品移库单(一步)修改为内调出入库单(一步)方法使用
  11. android列表停止滚动,加载图片,较为通用的一种办法
  12. 利用jdbc简单封装一个小框架(类似DBUtils)
  13. 详解 $().css(&#39;width&#39;)和$().width()的区别
  14. LeetCode 203. Remove Linked List Elements 移除链表元素 C++/Java
  15. 20165221 JAVA第一周学习心得及体会
  16. 【T09】要认识到TCP是一个可靠的,但不是绝对可靠的协议
  17. Redis数据结构之哈希
  18. Linux文件的所有权与权限
  19. 《js高级程序设计》--第三章数据类型
  20. Oracle ASM 相关的 视图(V$) 和 数据字典(X$)

热门文章

  1. tomcat部署maven web项目
  2. hdu5628 Clarke and math
  3. P4592 [TJOI2018]异或
  4. (转载)windows下安装配置Xampp
  5. Unity3D学习笔记(三十七):顶点偏移和溶解
  6. 操作 html 的时候是使用 dom 方法还是字符串拼接?
  7. js 字符串加密解密
  8. 最简单的服务器和客户机(python3的编码与解码问题)
  9. 【Ruby】【目录 &amp; 引用 &amp; 文件 】
  10. Vue运行报错--not defined