【模板】KMP與MP的區別(洛谷P3375)

學KMP的時候巨佬說我這寫的是MP,仔細去查了查資料,才發現了區別。 洛谷這道題用KMP是解決不了的,KMP的nxt數組和MP的nxt數組略有不一樣。 http://www.javashuo.com/article/p-sqquamym-gz.html 這位大佬寫的過程很清楚,可是程序不是怎麼看得懂,並且實際放到洛谷的題上測應該也是過不了的(應該。 直接放程序8html

#include<bits/stdc++.h>
#define rep(i,k,n) for(int i=k;i<=n;i++)
#define pb push_back
using namespace std;
const int maxn=1000005;
int n,m,Knxt[maxn],Mnxt[maxn];
char str[maxn],ptr[maxn];
vector<int> ans;
void KMP_cal_nxt()
{
	int k=-1;Knxt[0]=-1;
	rep(i,1,m-1)
	{
		while(k>-1 && ptr[k+1]!=ptr[i]) k=Knxt[k];
		if(ptr[k+1]==ptr[i]) k++;
		if(ptr[k+1]==ptr[i+1]) Knxt[i]=Knxt[k];
		else Knxt[i]=k;
	}
}
void MP_cal_nxt()
{
	int k=-1;Mnxt[0]=-1;
	rep(i,1,m-1)
	{
		while(k>-1 && ptr[k+1]!=ptr[i]) k=Mnxt[k];
		if(ptr[k+1]==ptr[i]) k++;
		Mnxt[i]=k;
	}
}
void KMP()
{
	KMP_cal_nxt();
	int k=-1;
	rep(i,0,n-1)
	{
		while(k>-1 && ptr[k+1]!=str[i]) k=Knxt[k];
		if(ptr[k+1]==str[i]) k++;
		if(k==m-1) ans.pb(i-m+2),k=Knxt[k];
	}
}
void MP()
{
	MP_cal_nxt();
	int k=-1;
	rep(i,0,n-1)
	{
		while(k>-1 && ptr[k+1]!=str[i]) k=Mnxt[k];
		if(ptr[k+1]==str[i]) k++;
		if(k==m-1) ans.pb(i-m+2),k=Mnxt[k];
	}
}

int main()
{
	scanf("%s%s",str,ptr);
	n=strlen(str);
	m=strlen(ptr);
	MP_cal_nxt();KMP();//用了MP的nxt數組,題目要求,可是是用KMP函數匹配
	rep(i,0,(int)ans.size()-1)
	printf("%d\n",ans[i]);
	rep(i,0,m-1) printf("%d ",Mnxt[i]+1);
	return 0;
}
相關文章
相關標籤/搜索