hihoCoder #1015 : KMP算法

忘是亡心i 2022-05-30 06:24 66阅读 0赞

时间限制: 1000ms

单点时限: 1000ms

内存限制: 256MB

### 描述 ###

小Hi和小Ho是一对好朋友，出生在信息化社会的他们对编程产生了莫大的兴趣，他们约定好互相帮助，在编程的学习道路上一同前进。

这一天，他们遇到了一只河蟹，于是河蟹就向小Hi和小Ho提出了那个经典的问题：“小Hi和小Ho，你们能不能够判断一段文字（原串）里面是不是存在那么一些……特殊……的文字（模式串）？”

小Hi和小Ho仔细思考了一下，觉得只能想到很简单的做法，但是又觉得既然河蟹先生这么说了，就肯定不会这么容易的让他们回答了，于是他们只能说道：“抱歉，河蟹先生，我们只能想到时间复杂度为（文本长度 \* 特殊文字总长度）的方法，即对于每个模式串分开判断，然后依次枚举起始位置并检查是否能够匹配，但是这不是您想要的方法是吧？”

河蟹点了点头，说道：”看来你们的水平还有待提高，这样吧，如果我说只有一个特殊文字，你能不能做到呢？“

小Ho这时候还有点晕晕乎乎的，但是小Hi很快开口道：”我知道！这就是一个很经典的模式匹配问题！可以使用KMP算法进行求解！“

河蟹满意的点了点头，对小Hi说道：”既然你知道就好办了，你去把小Ho教会，下周我有重要的任务交给你们！“

”保证完成任务！”小Hi点头道。

[提示一：KMP的思路][KMP]

[提示二：NEXT数组的使用][KMP]

[提示三：如何求解NEXT数组][KMP]

### 输入 ###

第一行一个整数N，表示测试数据组数。

接下来的N\*2行，每两行表示一个测试数据。在每一个测试数据中，第一行为模式串，由不超过10^4个大写字母组成，第二行为原串，由不超过10^6个大写字母组成。

其中N<=20

### 输出 ###

对于每一个测试数据，按照它们在输入中出现的顺序输出一行Ans，表示模式串在原串中出现的次数。

样例输入

5
    HA
    HAHAHA
    WQN
    WQN
    ADA
    ADADADA
    BABABB
    BABABABABABABABABB
    DAD
    ADDAADAADDAAADAAD

样例输出

3
    1
    3
    1

题目解析:

这个题目非常的简单,就是直接的kmp模板题目,很水,1A

代码:

#include <iostream>
    #include <cstdio>
    #include <cstring>
    using namespace std;
    
    const int N = 1000002;
    int next6[N];
    char S[N], T[N];
    int slen, tlen;
    
    void getnext6()
    {
        int j, k;
        j = 0; k = -1; next6[0] = -1;
        while(j < tlen)
            if(k == -1 || T[j] == T[k])
                next6[++j] = ++k;
            else
                k = next6[k];
    
    }
    /*
    返回模式串在主串S中出现的次数
    */
    int KMP_Count()
    {
        int ans = 0;
        int i, j = 0;
    
        if(slen == 1 && tlen == 1)
        {
            if(S[0] == T[0])
                return 1;
            else
                return 0;
        }
        getnext6();
        for(i = 0; i < slen; i++)
        {
            while(j > 0 && S[i] != T[j])
                j = next6[j];
            if(S[i] == T[j])
                j++;
            if(j == tlen)
            {
                ans++;
                j = next6[j];
            }
        }
        return ans;
    }
    int main()
    {
        //freopen("in.txt","r",stdin);
        int TT;
        int i, cc;
        cin>>TT;
        while(TT--)
        {
            cin>>T>>S;
            slen = strlen(S);
            tlen = strlen(T);
            //cout<<"模式串T在主串S中首次出现的位置是: "<<KMP_Index()<<endl;
            cout<<KMP_Count()<<endl;
        }
        return 0;
    }

[KMP]: http://hihocoder.com/problemset/problem/1015#