【ZCMU1953】子串查找（KMP模板题）

题目链接

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;
}