KMP(acwing)

831. KMP字符串

•   题目 
•   提交记录 
•   讨论 
•   题解 
•   视频讲解 

给定一个字符串 S

，以及一个模式串 P

，所有字符串中只包含大小写英文字母以及阿拉伯数字。

模式串 P

在字符串 S

中多次作为子串出现。

求出模式串 P

在字符串 S

中所有出现的位置的起始下标。

输入格式

第一行输入整数 N

，表示字符串 P

的长度。

第二行输入字符串 P

。

第三行输入整数 M

，表示字符串 S

的长度。

第四行输入字符串 S

。

输出格式

共一行，输出所有出现位置的起始下标（下标从 0

开始计数），整数之间用空格隔开。

数据范围

1≤N≤105

1≤M≤106

输入样例：

3
aba
5
ababa

输出样例：

0 2

 

#include <iostream>
#include <string>

using namespace std;

const int maxn =1e6+10;
int ne[maxn];
char s[maxn],p[maxn];

void get_ne(int n)
{
    ne[0]=-1;
    for(int i=1,j=-1;i<n;++i)
    {
        while(j!=-1 && p[i]!=p[j+1])
            j=ne[j];
        if(p[i]==p[j+1])
            ++j;
        ne[i]=j;
    }
}

void KMP(int n,int m)
{
    get_ne(n);
    
    for(int i=0,j=-1;i<m;++i)
    {
        while(j!=-1 && s[i]!=p[j+1])
            j=ne[j];
            
        if(s[i]==p[j+1])
            ++j;
        if(j == n-1)
        {
            cout << i-n+1 << ' ';
            j=ne[j];
        }
    }
    
}

int main()
{
    int n;
    cin >> n;
    getchar();
    
    scanf("%s",p);
    int m;
    cin >> m;
    getchar();
    scanf("%s",s);
    KMP(n,m);
    
    return 0;
}