P3375 【模板】KMP

链接：P3375 【模板】KMP - 洛谷

题目描述

给出两个字符串 s1​ 和 s2​，若 s1​ 的区间 [l,r] 子串与 s2​ 完全相同，则称 s2​ 在 s1​ 中出现了，其出现位置为 l。
现在请你求出 s2​ 在 s1​ 中所有出现的位置。

定义一个字符串 s 的 border 为 s 的一个非 s 本身的子串 t，满足 t 既是 s 的前缀，又是 s 的后缀。
对于 s2​，你还需要求出对于其每个前缀 s′ 的最长 border t′ 的长度。

输入格式

第一行为一个字符串，即为 s1​。
第二行为一个字符串，即为 s2​。

输出格式

首先输出若干行，每行一个整数，按从小到大的顺序输出 s2​ 在 s1​ 中出现的位置。
最后一行输出 ∣s2​∣ 个整数，第 i 个整数表示 s2​ 的长度为 i 的前缀的最长 border 长度。

输入输出样例

输入 #1复制运行

ABABABC
ABA

输出 #1复制运行

1
3
0 0 1 

说明/提示

样例 1 解释

。

对于 s2​ 长度为 3 的前缀 ABA，字符串 A 既是其后缀也是其前缀，且是最长的，因此最长 border 长度为 1。

数据规模与约定

本题采用多测试点捆绑测试，共有 3 个子任务。

• Subtask 1（30 points）：∣s1​∣≤15，∣s2​∣≤5。
• Subtask 2（40 points）：∣s1​∣≤104，∣s2​∣≤102。
• Subtask 3（30 points）：无特殊约定。

对于全部的测试点，保证 1≤∣s1​∣,∣s2​∣≤106，s1​,s2​ 中均只含大写英文字母。

C++代码如下：

#include<bits/stdc++.h>
using namespace std;
int kmp[1000005];
void fun(string a,string b)
{
    int lena=a.size();
    int lenb=b.size();
    int pos=0;
    for(int i=1;i<lenb;i++){
        while(pos&&b[i]!=b[pos]){
            pos=kmp[pos-1];
        }
        if(b[pos]==b[i]){
            pos++;
        }
        kmp[i]=pos;
    }
    pos=0;
    for(int i=0;i<lena;i++){
        while(pos&&a[i]!=b[pos]){
            pos=kmp[pos-1];
        }
        if(b[pos]==a[i]){
            pos++;
        }
        if(pos==lenb){
            cout<<i-lenb+1+1<<endl;
            pos=kmp[pos-1];
        }
    }
    for(int i=0;i<lenb;i++){
        cout<<kmp[i]<<" ";
    }
}
int main()
{
    string a,b;
    cin>>a>>b;
    fun(a,b);
    return 0;
}