开发: C++知识库 Java知识库 JavaScript Python PHP知识库人工智能区块链大数据移动开发嵌入式开发工具数据结构与算法开发测试游戏开发网络协议系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑笔记本显卡显示器固态硬盘硬盘耳机手机 iphone vivo oppo 小米华为单反装机图拉丁

-> C++知识库 -> C语言KMP算法 - strstr strrstr 实现 -> 正文阅读

[C++知识库]C语言KMP算法 - strstr strrstr 实现

C语言KMP算法实现

使用到的头文件

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

strstr 实现

char *strstr(const char *srcStr, const char *subStr)
{
        if (!srcStr || !subStr)  return (char *)NULL;
        const int subSize = strlen(subStr);
        int *next = malloc(sizeof(int) * subSize);
        if (next == NULL)  return (char *)NULL;
        int i = 0;
        int j = -1;
        next[i] = j;
        while (i < subSize - 1)
        {
                if ((j == -1) || (subStr[i] == subStr[j]))
                        next[i] = (subStr[++i] == subStr[++j]) ? next[j] : j;
                else
                        j = next[j];
        }
        i = j = 0;
        const int srcSize = strlen(srcStr);
        while (i < srcSize && j < subSize)
        {
                j = (j == -1 || srcStr[i] == subStr[j]) ? (i++, j + 1) : next[j];
        }
        if (next != NULL)
        {
                free(next);
                next = NULL;
        }
        return (char *)((j == subSize) ? (srcStr + i - subSize) : NULL);
}

strrstr 实现

char *strrstr(const char *srcStr, const char *subStr)
{
        if (!srcStr || !subStr)  return (char *)NULL;
        const int srcSize = strlen(srcStr);
        const int subSize = strlen(subStr);
        int i = subSize - 1;
        int j = subSize;
        int *prefix = malloc(sizeof(int) * subSize);
        if (prefix == NULL)  return (char *)NULL;
        prefix[i] = j;
        while (i > 0)
        {
                if ((j == subSize) || (subStr[i] == subStr[j]))
                        prefix[i] = (subStr[--i] == subStr[--j]) ? prefix[j] : j;
                else
                        j = prefix[j];
        }
        i = srcSize - 1, j = subSize - 1;
        while (i >= 0 && j >= 0)
        {
                j = (j == subSize || srcStr[i] == subStr[j]) ? (i--, j - 1) : prefix[j];
        }
        if (prefix != NULL)
        {
                free(prefix);
                prefix = NULL;
        }
        return (char *)((j == -1) ? (srcStr + i + 1) : NULL);
}