Educational Codeforces Round 134 (Rated for Div. 2) E. Prefix Function Queries(KMP优化)

最新推荐文章于 2026-06-17 20:26:24 发布

原创最新推荐文章于 2026-06-17 20:26:24 发布 · 381 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#算法

CF 专栏收录该内容

17 篇文章

订阅专栏

本文介绍了一种针对KMP算法的优化方法，通过预处理字符串并构建特殊数组来提高匹配效率。这种方法尤其适用于需要频繁查询的情况，能显著减少运行时间。

题意：

给定一个 $s$ 串，然后 $q$ 个询问，每个询问给个 $t$ 串（保证 $t$ 串长度<=10）然后把 $t$ 串接到 $s$ 串后面，然后求t中的字符的 $n e x t$ 数组（每个询问独立，也就是说每个询问后 $s$ 串还是原本的 $s$ 串，不会变成 $s + t$ ）

题解：

先预处理 $s$ 串的 $n e x t$ 数组，然后对于每个询问对结尾 $t$ 串跑 $n e x t$ 数组。这是显然的做法，然后TLE了。只要 $t$ 串的数据是正好要跳 $s$ 串整个长串的话，每次就是跑 $1 e 6$ 的长度了，这样的话极限时间复杂度是 $O (q * ∣ s ∣)$ ,所以会TLE。

然后就是想怎么优化这个 $km p$ 。 $km p$ 的本质是一颗树（如果每个 $n e x t [i]$ 向 $i$ 建边，那么会形成一颗以 $0$ 为根的树）,然后 $km p$ 匹配的过程就是一直往父节点跳的过程。那么如果能够跳一次就直接跳到应该匹配的位置就好了，而不是一直跳，然后边跳边判断。

所以引入一个 $c h [N] [26]$ 数组。 $c h [i] [j]$ 表示 $i$ 这个点若匹配字符 $j$ 要会匹配到的位置

那么原先，next[i]匹配过程为：

void get_next(string &s) {
    nxt[1] = 0;//下标从1开始
    for (int i = 2; i < s.length(); i++) {
        nxt[i] = nxt[i - 1];
        while (nxt[i] && s[i] != s[nxt[i] + 1])
            nxt[i] = nxt[nxt[i]];
        if (s[nxt[i] + 1] == s[i])nxt[i]++;
    }
}

就变为了

nxt[1] = 0;ch[0][s[1] - 'a'] = 1;
for (int i = 2; i <= n; i++) {
    nxt[i] = ch[nxt[i - 1]][s[i] - 'a'];
    ch[i - 1][s[i] - 'a'] = i;
    for (int j = 'a'; j <= 'z'; j++) {
        if (j == s[i])continue;
        ch[i - 1][j - 'a'] = ch[nxt[i - 1]][j - 'a'];
    }
}

然后附上严格鸽的题解：严格鸽的题解

.下面是我的关键代码


void solve(){
    cin>>s+1;
    n=strlen(s+1);
    nxt[1]=0;ch[0][s[1]-'a']=1;
    for(int i=2;i<=n;i++){
        int c=s[i]-'a';
        nxt[i]=ch[nxt[i-1]][c];
        ch[i-1][c]=i;
        for(int j='a';j<='z';j++){
            if(j==s[i]) continue;
            int c=j-'a';
            ch[i-1][c]=ch[nxt[i-1]][c];
        }
    }
    int m;cin>>m;
    while(m--){
        cin>>t+1;
        int len=strlen(t+1)+n;
        for(int i=n+1;i<=len;i++) s[i]=t[i-n];
        for(int i=n+1;i<=len;i++){
            int c=s[i]-'a';
            nxt[i]=ch[nxt[i-1]][c];
            ch[i-1][c]=i;
            for(int j='a';j<='z';j++){
                if(j==s[i]) continue;
                int c=j-'a';
                ch[i-1][c]=ch[nxt[i-1]][c];
            }
            cout<<nxt[i]<<' ';
        }cout<<endl;
    }
}