YbtOJ——KMP 算法【例题1】子串查找

A. 【例题1】子串查找

题目

终于学会了KMP !
板子就不解释了，贴上一篇别人的KMP好文：数据结构KMP算法配图详解（超详细）

代码

#include<iostream>
using namespace std;
string a,b;
int p[100001],n,ans;
void in()
{
	cin>>a>>b;
}
void pre(string s)//预处理最长相同前缀后缀
{
	p[0]=-1;
	p[1]=0;
	int j=0,k=-1;
	while(j<s.size()-1)
	{
		if(k==-1||s[j]==s[k])//若两字符相同，则两个比较字符同时前进
		{
			k++;
			j++;
			p[j]=k;
		}
		else k=p[k];//否则子串的比较字符返回最长相同前缀后缀的位置
	}
}
void kmp()
{
	pre(b);
	int i=-1,j=-1;
	while(i<0||i<a.size())
	{
		while(j>=0&&a[i+1]!=b[j+1]) j=p[j];//不断回溯直到两个比较字符相同
		if(a[i+1]==b[j+1])//若两字符相同，则两个比较字符同时前进
		{
			i++;
			j++;
		}
		else i++;
		if(j==b.size()-1)//若查找到一处答案，子串的比较字符回溯并继续查找。
		{
			ans++;
			j=p[j]; 
		}
	}

}
int main()
{
	in();
	kmp();
	cout<<ans;
}