KMP算法

详细参考

这个是用来加快cin和cout的

实例

给定一个模式串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>
using namespace std;

const int N = 100010,M = 1000010;
char a[N],b[M];
int ne[N]; //next数组
int n,m;
//下标从1开始存储
void cal_next(char str[],int len)
{
    ne[0] = 0;
    ne[1] = 0;
    for(int i=2;i<=len;i++)
    {
        int j = ne[i-1]; //j代表已经匹配的长度
        if(j!=0){
            while(j!=0)
            {
                if(str[j+1]==str[i]) 
                {
                    ne[i] = j+1;
                    break;
                }
                else{
                    j = ne[j];
                }
            }
        }
        if(j==0){
            if(str[i]==str[1]) ne[i] = 1;
            else ne[i] = 0;
        }
        
    }
}


int main()
{
    cin >> n >> a+1 >> m >> b+1;
    cal_next(a,n);
    
     //str2是模板串
     //两个串的下标都是从1开始的
     
    //j代表模式串已经匹配成功的字符个数
    //现在开始匹配模板串的第i号是否匹配
    // kmp算法实现
    
    for(int i=1,j=0;i<=m;i++)
    {
        while(j!=0 && b[i] != a[j+1]) j = ne[j];
        if(b[i]==a[j+1]) j++;
        if(j==n)
        {
            printf("%d ",i-n);
            j = ne[j];
        }
        
    }

    
    return 0;
    
}