zcmu-1953

1953: #103. 子串查找

Time Limit: 5 Sec Memory Limit: 256 MB
Submit: 112 Solved: 46
[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 仅包含大小写字母。

思路：题目要求的时间为5秒，这里可以想到srting类的find（）函数，时间上5秒够了，但是还是wa几次，没理解到find（）函数其实是在找子串的时候，从i个位置开始，比如i=100；要是没找到就跳出循环，因为接下去的再也找不到了，果然还是自己对函数理解不透彻。
另外一个做法是kmp算法，时间耗时小，比较推荐。kmp算法有模板。套一下模板就成了。

ac代码：
string的find（）函数；

#include<bits/stdc++.h>using namespace std;int main(){    string a;    string b;    while(cin>>a)    {        getchar();        cin>>b;        int n=a.size();        int m=b.size();        int i=0,count=0;        while(1)        {            i=a.find(b,i);            if(i==-1)break;            if(i!=-1)count++;            i++;        }        cout<<count<<endl;    }    return 0;}

kmp算法：#include<iostream>#include<cstdio>#include<cstring>using namespace std;const int N=1000001;char a[N],b[N];int la,lb;int p[N];int ans;void makep(){    int j=0;    p[0]=0;    for(int i=1;i<lb;i++)    {        while(j>0&&b[j]!=b[i])            j=p[j-1];        if(b[j]==b[i])            j++;        p[i]=j;    }}void KMP(){    int j=0;    for(int i=0;i<la;i++)    {        while(j>0&&b[j]!=a[i])            j=p[j-1];        if(b[j]==a[i])            j++;        if(j==lb)            ans++,            j=p[j-1];    }    printf("%d",ans);}int main(){    scanf("%s %s",a,b);    la=strlen(a);lb=strlen(b);    makep();    KMP();    return 0;}