#103. 子串查詢

自為風月馬前卒發表於2017-07-12
記憶體限制：256 MiB時間限制：500 ms標準輸入輸出
題目型別：傳統評測方式：文字比較
上傳者： 匿名
提交提交記錄統計討論測試資料
 

題目描述

這是一道模板題。

給定一個字串 A AA 和一個字串 B BB，求 B BB 在 A AA 中的出現次數。

A AA 中不同位置出現的 B BB 可重疊。

輸入格式

輸入共兩行，分別是字串 A AA 和字串 B BB。

輸出格式

輸出一個整數，表示 B BB 在 A AA 中的出現次數。

樣例

樣例輸入

zyzyzyz
zyz

樣例輸出

3

資料範圍與提示

1≤A,B 1 \leq A, B1≤A,B 的長度 ≤106 \leq 10 ^ 6≤10​6​​，A AA、B BB 僅包含大小寫字母。

 

裸的KMP。

發現一前寫錯了一個地方

第40行的while一直寫的是if。

但是在洛谷能AC，

估計是資料渣的一逼，，

 

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int MAXN=100000001;
inline void read(int &n)
{
	char c='+';int x=0;bool flag=0;
	while(c<'0'||c>'9')
	{c=getchar();if(c=='-')flag=1;}
	while(c>='0'&&c<='9')
	{x=(x<<1)+(x<<3)+c-48,c=getchar();}
	flag==1?n=-x:n=x;
}
char a[MAXN];
char b[MAXN];
int la;
int lb;
int p[MAXN];
int ans=0;
void makep()
{
	int j=0;
	for(int i=1;i<lb;i++)
	{
		while(b[i]!=b[j]&&j>0)
			j=p[j-1];
		if(b[i]==b[j])
			j++;
		p[i]=j;
	}
}
void kmp()
{
	int j=0;
	for(int i=0;i<la;i++)
	{
		while(a[i]!=b[j]&&j>0)
			j=p[j-1];
		if(a[i]==b[j])
			j++;
		if(j==lb)
			ans++;
	}
	printf("%d",ans);
}
int main()
{
	scanf("%s%s",a,b);
	la=strlen(a);
	lb=strlen(b);
	makep();
	kmp();
	return 0;
}

　　

 

相關文章