AcWing
  • 首页
  • 课程
  • 题库
  • 更多
    • 竞赛
    • 题解
    • 分享
    • 问答
    • 应用
    • 校园
  • 关闭
    历史记录
    清除记录
    猜你想搜
    AcWing热点
  • App
  • 登录/注册

AcWing 3011. 甲苯先生和大中锋的字符串    原题链接    困难

作者: 作者的头像   Conan15 ,  2025-04-16 14:54:30 · 福建 ,  所有人可见 ,  阅读 35


1


后缀数组笔记。

首先只要学过 SA 就并不难刻画“恰好出现 $k$ 次”这一条件。
设 $u$ 为 $[l,r]$ height 的 $\min$,且 $r-l+1$ 即区间长度为定值 $k-1$,这个可以用单调队列简单维护。
则 $[1,u]$ 的前缀出现次数显然不小于 $k$,至于恰好为 $k$ 只需要钦定 $d = \max(ht_{l-1},ht_{r+1})$ 就行了,合法的长度区间为 $[d+1,u]$,相当于这个区间出现次数 $+1$。
现在需要一个数据结构支持区间 $+1$ 和全局查询最大值。由于查询在加法之后因此可以用差分。

对于 $k=1$,由于单调队列区间长度为 $k-1=0$,需要特殊处理它的 $u$ 为 $n-sa_i+1$ 即后缀长度。

#include <bits/stdc++.h>
using namespace std;
const int N = 2e5 + 5;
int T, n, k;
char s[N];

int x[N], y[N], rk[N], sa[N], cnt[N];
void init_SA() {
    int v = 128;
    for (int i = 0; i <= v; i++) cnt[i] = 0;
    for (int i = 1; i <= n; i++) ++cnt[x[i] = s[i]];
    for (int i = 1; i <= v; i++) cnt[i] += cnt[i - 1];
    for (int i = n; i >= 1; i--) sa[ cnt[x[i]]-- ] = i;
    for (int len = 1; ; len <<= 1) {
        int tot = 0;
        for (int i = n - len + 1; i <= n; i++) y[++tot] = i;
        for (int i = 1; i <= n; i++)
            if (sa[i] > len) y[++tot] = sa[i] - len;

        for (int i = 0; i <= v; i++) cnt[i] = 0;
        for (int i = 1; i <= n; i++) ++cnt[x[i]];
        for (int i = 1; i <= v; i++) cnt[i] += cnt[i - 1];
        for (int i = n; i >= 1; i--) sa[ cnt[x[y[i]]]-- ] = y[i];

        swap(x, y), tot = 0;
        for (int i = 1; i <= n; i++) {
            if (y[sa[i]] == y[sa[i - 1]] && y[sa[i] + len] == y[sa[i - 1] + len]) x[sa[i]] = tot;
            else x[sa[i]] = ++tot;
        }
        v = tot;
        if (v == n) break;
    }
}

int ht[N];
void init_height() {
    for (int i = 1; i <= n; i++) rk[sa[i]] = i;
    for (int i = 1, j = 0, now = 0; i <= n; i++) {
        if (rk[i] == 1) { ht[rk[i]] = 0; continue; }
        j = sa[rk[i] - 1], now = max(now - 1, 0);
        while (i + now <= n && j + now <= n && s[i + now] == s[j + now]) now++;
        ht[rk[i]] = now;
    }
}

int q[N], st, ed;
int b[N];
void solve() {
    for (int i = 1; i < k; i++) {
        while (st <= ed && ht[q[ed]] > ht[i]) ed--;
        q[++ed] = i;
    }
    for (int i = k; i <= n; i++) {
        while (st <= ed && q[st] <= i - k + 1) st++;
        while (st <= ed && ht[q[ed]] > ht[i]) ed--;
        q[++ed] = i;
        int l = ht[i - k + 1], r = ht[i + 1];
        int u = (k == 1) ? (n - sa[i] + 1) : ht[q[st]], d = max(l, r) + 1;
        if (d <= u) b[d]++, b[u + 1]--;
    }
    int Max = 0, id = 0;
    for (int i = 1; i <= n; i++) {
        b[i] += b[i - 1];
        if (Max <= b[i]) Max = b[i], id = i;
    }
    printf("%d\n", (Max == 0) ? -1 : id);
}

void clr() {
    for (int i = 0; i <= ed; i++) q[i] = 0; st = 1, ed = 0; //clr
    for (int i = 0; i <= n + 1; i++) b[i] = sa[i] = rk[i] = ht[i] = 0;
}

void mian() {
    scanf("\n %s %d", s + 1, &k), n = strlen(s + 1);
    clr();
    init_SA();
    init_height();
    solve();
}

int main() {
    scanf("%d", &T);
    while (T--) mian();
    return 0;
}

0 评论

App 内打开
你确定删除吗?
1024
x

© 2018-2025 AcWing 版权所有  |  京ICP备2021015969号-2
用户协议  |  隐私政策  |  常见问题  |  联系我们
AcWing
请输入登录信息
更多登录方式: 微信图标 qq图标 qq图标
请输入绑定的邮箱地址
请输入注册信息