poj_3461 kmp

题目大意

给定两个字符串S1， S2，求S1中包含多少个S2串。其中S1长度最大为 1000000， S2长度最大为10000。

题目分析

典型的字符串匹配问题，直接匹配显然会超时，考虑使用kmp算法。
需要注意的是，尽量减少求串长度的strlen的执行次数。

实现(c++)

#define _CRT_SECURE_NO_WARNINGS
#define MAX_WORD_LEN 10005
#define MAX_TEXT_LEN 1000005
#include<stdio.h>
#include<string.h>

char gWord[MAX_WORD_LEN];
char gText[MAX_TEXT_LEN];
int gNext[MAX_WORD_LEN];
void GenerateNext(const char* sub_str, int* next, int len){
next[0] = 0;								//next[i]表示sub_str[0, i]中最长相同前后缀的长度
for (int i = 1; i < len; i++){
int j = next[i - 1];
while (sub_str[j] != sub_str[i]){		//不断回溯，每次利用next数组来确定下一个需要判断的位置
if (j == 0){
j = -1;
break;
}
j = next[j - 1];
}
if (j < 0)
next[i] = 0;
else
next[i] = j + 1;
}
}

//返回母串中有多少个子串
int Kmp(char* mas_str, const char* sub_str, int* next, int len){
int count = 0;
char* pos = mas_str;	//母串中当前检查的位置
int i = 0;				//子串中当前检查的位置
char* end = mas_str + strlen(mas_str);
while (pos + len - i <= end){
while (i < len && *(pos + i) == sub_str[i]){
i++;
}
if (i == len){
count++;
}
if (i == 0){	//i == 0时，没有next[i-1]，单独处理
pos++;
}
else{
pos += (i - next[i - 1]);	//利用next数组进行更新
i = next[i - 1];
}
}
return count;
}

int main(){
int cas;
scanf("%d", &cas);
getchar();
for (int i = 0; i < cas; i++){
scanf("%s", gWord);
scanf("%s", gText);
int len = strlen(gWord);
GenerateNext(gWord, gNext, len);
int count = Kmp(gText, gWord, gNext, len);
printf("%d\n", count);
}
return 0;
}