义冢OJ P5018 Oulipo

义冢 OJ P5018 Oulipo#

描述#

给出兩個字串 W 和 T，求 T 中有幾個 W 子字串

輸入#

第一行為資料組數.

每組資料有兩行 W 和 T，表示模式字串和原始字串.

輸出#

對每組資料，每行一個數，表示匹配數.

輸入範例 1#

3
BAPC
BAPC
AZA
AZAZAZA
VERDI
AVERDXIVYERDIAN

輸出範例 1#

1
3
0

提示#

1 ≤ |W| ≤ 10,000 (here |W| denotes the length of the string W). |W| ≤ |T| ≤ 1,000,000.

做#

KMP 板子題

只能寫寫板子來水水博客了。。。。

Code#

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;

const int MAXN=1e6+5;

int nxt[MAXN]={0};

void setNext(string s);
int kmp(string s1,string s2);

int n;

int main()
{
	cin>>n;
	string s1,s2;
	for(int i=1;i<=n;++i)
	{
		cin>>s1>>s2;
		setNext(s1);
		cout<<kmp(s2,s1)<<endl;
	}
	
	return 0;
}
void setNext(string s)
{
	memset(nxt,0,sizeof(nxt));
	nxt[0]=-1;
	int sl=s.size();
	for(int i=1;i<sl;++i)
	{
		int t=nxt[i-1];
		while(s[t+1]!=s[i]&&t>=0)
		{
			t=nxt[t];
		}
		if(s[t+1]==s[i])
		{
			nxt[i]=t+1;
		}
		else
		{
			nxt[i]=-1;
		}
	}
}
int kmp(string s1,string s2)
{
	int i=0,j=0;
	int sl1=s1.size(),sl2=s2.size();
	int cnt=0;
	while(i<sl1)
	{
		if(s1[i]==s2[j])
		{
			++i;
			++j;
			if(j==sl2)
			{
				++cnt;
				j=nxt[j-1]+1;
			}
		}
		else
		{
			if(j==0)++i;
			else j=nxt[j-1]+1;
		}
	}
	return cnt;
}