1953: #103. 子串查找【string:: find()及KMP算法】

1953: #103. 子串查找

Time Limit: 5 Sec  Memory Limit: 256 MB
Submit: 192  Solved: 79
[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 的长度 ≤10^6 ，A 、B 仅包含大小写字母。

题解：第一次提交的时候是用了string中的find()函数，运行时间比较长，find函数是从第i个位置开始查找，如果找到就会返回第一个出现的位置，如果没找到就返回-1

#include<bits/stdc++.h>
using namespace std;

int main(void){
	string a,b;
	while(cin>>a){
		getchar();//取消换行对后面的string的影响
		cin>>b;
		int i=0,ans=0;
		while(1){
			i=a.find(b,i);//从i位置开始，寻找第一个出现b的位置 
			if(i==-1)
			break;
			if(i!=-1) ans++;
			i++;
		}
		printf("%d\n",ans);
	}
	return 0;
} 

KMP算法（模板）：

#include <bits/stdc++.h>
using namespace std;
const int N = 1000005;
char S[N], T[N];
int Next[N];
int slen, tlen;
void getNext()
{
    slen = strlen(S);
    tlen = strlen(T);
    int j=0,k=-1;
    Next[0] = -1;
    while(j<tlen)
        if(k==-1||T[j]==T[k])
            Next[++j]=++k;
        else
            k=Next[k];
}
int KMP_Count()
{
    int ans=0,i,j=0;
    getNext();
    for(i=0;i<slen;i++)
    {
        while(j>=0&&S[i]!=T[j])
            j=Next[j];
        if(j==-1||S[i]==T[j])
            j++;
        if(j==tlen)
        {
            ans++;
            j=Next[j];
        }
    }
    return ans;
}
int main()
{
    while(~scanf("%s%s",S,T))
        cout<<KMP_Count()<<endl;
    return 0;
}