hdu2203（KMP模板）

最新推荐文章于 2022-08-16 16:03:58 发布

丿Smile灬晨星

最新推荐文章于 2022-08-16 16:03:58 发布

阅读量540

点赞数

CC 4.0 BY-SA版权

分类专栏：算法模板字符串

本文链接：https://blog.youkuaiyun.com/cxsmile/article/details/49012365

算法模板同时被 2 个专栏收录

14 篇文章

订阅专栏

字符串

6 篇文章

订阅专栏

本文详细介绍了KMP算法的基本原理及其在字符串匹配任务中的优化实现，并通过实例展示了如何利用KMP算法高效地进行模式查找。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

参考代码：

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<vector>
#include<set>
#include<map>
#include<queue>
#include<sstream>
#include<string>
#include<bitset>
using namespace std;

typedef long long LL;

const LL LINF = (1LL <<63);
const int INF = 1 << 31;


const int NS = 200010;
const int MS = 19;
const int MOD = 1000000007;

char s1[NS], s2[NS];
int snext[NS];

void getNext(char* pattern, int* next)
{
    //未优化的next数组，字符下标从0开始
    int len = strlen(pattern);
    int j = -1;
    next[0] = -1;
    for(int i = 0; i < len;)
    {
        while(j >= 0 && pattern[i] != pattern[j])
        {
            j = next[j];
        }
        i++, j++;
        next[i] = j;
    }
}

void getOptNext(char* pattern, int* next)
{
    //优化的next数组，字符下标从0开始
    int len = strlen(pattern);
    int j = -1;
    next[0] = -1;
    for(int i = 0; i < len;)
    {
        while(j >= 0 && pattern[i] != pattern[j])
        {
            j = next[j];
        }
        i++, j++;
        if(pattern[i] != pattern[j])
        {
            next[i] = j;
        }
        else
        {
            next[i] = next[j];
        }
    }
}

bool KMP(char* target, char* pattern, int* next)
{
    int n = strlen(target);
    int m = strlen(pattern);

    int j = 0;
    for(int i = 0; i < n;)
    {
        if(target[i] == pattern[j])
        {
            i++, j++;
            if(m == j)
            {
                return true;
            }
        }
        else
        {
            j = next[j];
            if(j < 0)
            {
                i++, j++;
            }
        }
    }
    return false;
}

int main()
{
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
//    freopen("out.txt", "w", stdout);
#endif

    while(~scanf("%s %s", s1, s2))
    {
        int n = strlen(s1);
        for(int i = 0; i < n; i++)
        {
            s1[i + n] = s1[i];
        }
        n <<= 1;
        s1[n] = '\0';
        getOptNext(s2, snext);
        bool ans = KMP(s1, s2, snext);

        if(ans)
        {
            puts("yes");
        }
        else
        {
            puts("no");
        }
    }

    return 0;
}