831. KMP字符串

831. KMP字符串

难

题意

给定一个字符串 SS，以及一个模式串 PP，所有字符串中只包含大小写英文字母以及阿拉伯数字。

模式串 PP 在字符串 SS 中多次作为子串出现。

求出模式串 PP 在字符串 SS 中所有出现的位置的起始下标。

输入格式

第一行输入整数 NN，表示字符串 PP 的长度。

第二行输入字符串 PP。

第三行输入整数 MM，表示字符串 SS 的长度。

第四行输入字符串 SS。

输出格式

共一行，输出所有出现位置的起始下标（下标从 00 开始计数），整数之间用空格隔开。

数据范围

1≤N≤1051≤N≤105
1≤M≤1061≤M≤106

输入样例：

3
aba
5
ababa

输出样例：

0 2

题解

#include<iostream>
using namespace std;
const int N = 1e6+10 , M = 1e7+10;

char q[N] , s[M];

int ne[N] ;
 
int main(){
	int n, m;
	cin>>n>>q+1>>m>>s+1;
	for(int i = 2 , j = 0 ;  i <= n ; i++)
	{
		while(j&&q[i]!=q[j+1] ){
			j = ne[j];
			
			//TODO
		}
		if(q[i] == q[ j+1 ])j ++;
		ne[i] = j ;
	} 
	for(int i = 1 , j = 0 ; i <= m ; i++)
	{
		while(j&&s[i]!=q[j+1]){
			j=ne[j];//TODO
		}
		if(s[i] == q[j+1])j++;
		if(j == n)
		{
			cout<<i-j<<" ";
			j = ne[j] ;
		}
	}
}