洛谷P3375-KMP字符串匹配（模板）

题目链接：点击进入

思路

套模板（不用优化next，同时next多往后求一位）

代码

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=1e7+10;
int Next[maxn];
void getnext(string s)
{
	int len=s.size();
	Next[0]=-1;
	int k=-1,j=0;
	while(j<len)
	{
		if(k==-1||s[k]==s[j])
		{
			k++;
			j++;
			Next[j]=k;
		}
		else k=Next[k];
	}
}
void KMP(string s1,string s2)
{
	int l1=s1.size();
	int l2=s2.size();
	int i=0,j=0;
	while(i<l1)
	{
		if(j==-1||s1[i]==s2[j])
		{
			i++;
			j++;
		}
		else j=Next[j];
		if(j==l2)
		{
			cout<<i-j+1<<endl;
			j=Next[j];
		}
	}
}
int main()
{
	string s1,s2;
	cin>>s1>>s2;
	getnext(s2);
	KMP(s1,s2);
	for(int i=1;i<=s2.size();i++)
	cout<<Next[i]<<" ";
	return 0;
}

KMP模板（带优化）

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=1e7+10;
int Next[maxn];
void getnext(string s)
{
	int len=s.size();
	Next[0]=-1;
	int k=-1,j=0;
	while(j<len-1)
	{
		if(k==-1||s[k]==s[j])
		{
			k++;
			j++;
			if(s[j]!=s[k]) Next[j]=k;
			else Next[j]=Next[k];
		}
		else k=Next[k];
	}
}
void KMP(string s1,string s2)
{
	int l1=s1.size();
	int l2=s2.size();
	int i=0,j=0;
	while(i<l1&&j<l2)
	{
		if(j==-1||s1[i]==s2[j])
		{
			i++;
			j++;
		}
		else j=Next[j];
	}
	if(j==l2) cout<<i-j<<endl;
	else cout<<-1<<endl;
}
int main()
{
	string s1,s2;
	cin>>s1>>s2;
	getnext(s2);
	KMP(s1,s2);
	return 0;
}