Implement strStr()

最新推荐文章于 2021-02-27 02:02:17 发布

yxcbluesky

最新推荐文章于 2021-02-27 02:02:17 发布

阅读量237

点赞数

CC 4.0 BY-SA版权

分类专栏： leet code 文章标签：算法数据结构

本文链接：https://blog.youkuaiyun.com/yxcbluesky/article/details/39138275

leet code 专栏收录该内容

75 篇文章

订阅专栏

本文介绍了一种使用KMP算法实现字符串匹配的方法。通过计算前缀表并利用该表进行高效匹配，找到目标子串在主串中的首次出现位置。文章提供了详细的C++代码实现。

Implement strStr().

Returns a pointer to the first occurrence of needle in haystack, or null if needle is not part of haystack.

思路：这是非常典型的KMP算法题。

class Solution {
public:
    void computePrefix(char* needle, int* next)
    {
        int m = strlen(needle);
        int i = 0, j = -1;
        next[0] = -1;
        while(needle[i] != '\0')
        {
            while (j >= 0 && needle[j] != needle[i])
            {
                j = next[j];
            }       
            ++i, ++j;
            next[i] = j;
        }        
    }    
    char *strStr(char *haystack, char *needle) {
        if (strcmp(haystack, needle) == 0)
        {
            return haystack;
        }    
        int m = strlen(needle);
        int n = strlen(haystack);
        if (m == 0)
        {
            return haystack;
        }    
        int next[m];
        int i = 0, j = 0;
        computePrefix(needle, next);
        while(haystack[i] != '\0')
        {
            while(j >= 0 && haystack[i] != needle[j])
             j = next[j];
            ++i, ++j;
            if (needle[j] == '\0')
            {
                return haystack + i - j;
            }        
        }        
        return NULL;    
    }
};