洛谷-5410 【模板】扩展 KMP

最新推荐文章于 2025-04-20 10:50:37 发布

原创最新推荐文章于 2025-04-20 10:50:37 发布 · 187 阅读

0 ·

CC 4.0 BY-SA版权

算法题同时被 2 个专栏收录

310 篇文章

订阅专栏

模板

87 篇文章

订阅专栏

该博客介绍了如何使用扩展KMP算法解决计算两个字符串a和b的每一个后缀之间的最长公共前缀问题。输入包含两行，分别给出字符串a和b，输出包括两个数组，分别表示b的next数组和a的后缀与b的最长公共前缀长度。题目限制了字符串长度不超过100000。

题目描述
有两个字符串a,b,要求输出b与a的每一个后缀的最长公共前缀
输入格式
两行,分别为两个字符串a,b
输出格式
共两行
第一行有lenb个数,为b的next数组(特别地, $KaTeX parse error: Expected '}', got 'EOF' at end of input: next_{1$ }为lenb)
第二行有lena个数,即答案

输入输出样例
输入 #1
aaaabaa
aaaaa

输出 #1
5 4 3 2 1
4 3 2 1 0 2 1

说明/提示
lena<=100000
lenb<=100000

#include<bits/stdc++.h>
#define N 1000010
using namespace std;
int q,nxt[N],extend[N];
string s,t;
void getnxt(){
    nxt[0]=t.size();//nxt[0]一定是T的长度
    int now=0;
    while(t[now]==t[1+now]&&now+1<(int)t.size())now++;//这就是从1开始暴力
    nxt[1]=now;
    int p0=1;
    for(int i=2;i<(int)t.size();i++){
        if(i+nxt[i-p0]<nxt[p0]+p0)nxt[i]=nxt[i-p0];//第一种情况
        else{//第二种情况
            int now=nxt[p0]+p0-i;
            now=max(now,0);//这里是为了防止i>p的情况
            while(t[now]==t[i+now]&&i+now<(int)t.size())now++;//暴力
            nxt[i]=now;
            p0=i;//更新p0
        }
    }
}
void exkmp(){
    getnxt();
    int now=0;
    while(s[now]==t[now]&&now<min((int)s.size(),(int)t.size()))now++;//暴力
    extend[0]=now;
    int p0=0;
    for(int i=1;i<(int)s.size();i++){
        if(i+nxt[i-p0]<extend[p0]+p0)extend[i]=nxt[i-p0];//第一种情况
        else{//第二种情况
            int now=extend[p0]+p0-i;
            now=max(now,0);//这里是为了防止i>p的情况
            while(t[now]==s[i+now]&&now<(int)t.size()&&now+i<(int)s.size())now++;//暴力
            extend[i]=now;
            p0=i;//更新p0
        }
    }
}
int main(){
    cin>>s>>t;
    exkmp();
    int len=t.size();
    for(int i=0;i<len;i++)printf("%d ",nxt[i]);//输出nxt
    puts("");
    len=s.size();
    for(int i=0;i<len;i++)printf("%d ",extend[i]);//输出extend
    return 0;
}