1953: #103. 子串查找
Time Limit: 5 Sec Memory Limit: 256 MB
Submit: 193 Solved: 80
[Submit][Status][Web Board]
Description
这是一道模板题。
给定一个字符串 A 和一个字符串 B,求 B 在 A 中的出现次数。
A 中不同位置出现的 B 可重叠。
Input
输入共两行,分别是字符串 A 和字符串 B。
Output
输出一个整数,表示 B 在 A 中的出现次数。
Sample Input
zyzyzyz
zyz
Sample Output
3
HINT
1≤A,B 的长度 ≤106 ,A 、B 仅包含大小写字母。
解题思路:
KMP模板题,需要注意的是本题A 中不同位置出现的 B 可重叠。
代码:
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+5;
char a[maxn],b[maxn];
int next[maxn],ans=0;
void makenext()
{
int l=strlen(b);
int k,q;
next[0]=0;
for(q=1,k=0;q<l;q++)
{
while(k>0 && b[q]!=b[k])
k=next[k-1];
if(b[q]==b[k])
k++;
next[q]=k;
}
}
void kmp()
{
int n=strlen(a),m=strlen(b);
int i,j;
makenext();
for(i=0,j=0;i<n;i++)
{
while(j>0 && a[i]!=b[j])
j=next[j-1];
if(a[i]==b[j])
j++;
if(j==m)
{
ans++;
j=next[j-1];
printf("%d\n",j);
}
}
}
int main()
{
scanf("%s",a);
scanf("%s",b);
memset(next,0,sizeof(next));
kmp();
printf("%d\n",ans);
return 0;
}