Codeup100000634问题 A: 动物简介(animal)

题目描述：

到了动物园，琦琦开心得跳起来。哗，这里好多动物呀，有老虎，有狮子……，在开心之余，琦琦也不忘妈妈的教导：观察动物时要认真仔细，还要看动物园附上的动物简介呀。
动物的简介原来还有英文版的呢！为了卖弄自己的英文水平，琦琦就告诉妈妈每张动物简介里出现了多少次该动物的名称。注意：琦琦只认识小写字母，而且她只认得动物的单词，因此她认为monkeys或者smonkey或者smonkeys都是出现了monkey这个词。
你能编程完成琦琦的任务吗？

输入：

输入文件共n+2行：
第1行为数字n（n<=3000），表示该动物的简介共有n行。
第2行为一个单词，表示琦琦认识的动物名称。
接着是n行，每行为一个长度小于250个字符的字符串，表示动物的简介。

输出：

输出文件共1行，为简介里出现了多少次琦琦能识别出的动物的单词。

样例输入：

2
snake
The snake is a long and thin animal.
Snakes have no legs or feet.

样例输出：

1

实现代码：

#include<cstdio>
#include<string>
#include<iostream>
using namespace std;

const int maxn = 100 + 10;
int nextval[maxn];

void getNextval(string s) {
    int j = -1;
    nextval[0] = -1;
    for(int i = 1; i < s.size(); i++) {
        while(j != -1 && s[i] != s[j + 1]) {
            j = nextval[j];
        }
        if(s[i] == s[j + 1]) {
            j++;
        }
        if(j == -1 || s[i + 1] != s[j + 1]) {
            nextval[i] = j;
        } else {
            nextval[i] = nextval[j];
        }
    }
}

int KMP(string text, string pattern) {
    int n = text.size();
    int m = pattern.size();
    getNextval(pattern);
    int ans = 0, j = -1;
    for(int i = 0; i < n; i++) {
        while(j != -1 && text[i] != pattern[j + 1]) {
            j = nextval[j];
        }
        if(text[i] == pattern[j + 1]) {
            j++;
        }
        if(j == m - 1) {
            //  考虑单词在开头，结尾，中间三种情况
            ans++;
            j = -1;
        }
    }
    return ans;
}

int main() {
    int n;
    string pattern;
    string text;
    while(scanf("%d", &n) != EOF) {
        cin >> pattern;
        getchar();
        int num = 0;
        for(int i = 0; i < n; i++) {
            getline(cin, text);
            num += KMP(text, pattern);
        }
        printf("%d\n", num);
    }
    return 0;
}