[LeetCode] Implement strStr()

a83610312

于 2013-07-29 22:06:18 发布

阅读量958

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode

本文链接：https://blog.youkuaiyun.com/a83610312/article/details/9622723

LeetCode 专栏收录该内容

74 篇文章

订阅专栏

本文介绍了一种使用KMP算法实现字符串匹配的方法。通过C++代码详细展示了如何寻找一个字符串(needle)在另一个字符串(haystack)中首次出现的位置。如果找到则返回该位置的指针，否则返回null。代码中包含了初始化KMP前缀表的过程以及核心匹配逻辑。

Implement strStr():

Implement strStr().

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

KMP是基本功啊！但还是一下init里有个地方写错，牢记啊牢记！！

class Solution {
public:
    char *strStr(char *hay, char *need) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        
        if ( !hay || !need  )
	    	return NULL;
	    int nH=strlen(hay);
	    int nN=strlen(need);
	    if ( nN >nH )
		    return NULL;
	    if (nN==0)
		    return hay;

	    vector<int> pre(nN,-1);
	    init(pre,need,nN);
	    int pN=-1;
	    for(int pH=-1;pH<nH;pH++)
	    {
		    while(pN!=-1&&hay[pH+1]!=need[pN+1])
			    pN=pre[pN];
		    if(hay[pH+1]==need[pN+1])
			    pN++;
		    if ( pN == nN-1 )
			    return &hay[pH-nN+2];
	    }
	    return NULL;
    }
    void init(vector<int>& pre,char* s,int n)
    {
	    int p;
	    pre[0]=-1;
	    for(int i=1;i<n;i++)
	    {
		    p=pre[i-1];
		    while(p!=-1&&s[p+1]!=s[i])
			    p=pre[p];
		    if ( s[p+1]==s[i])
			    pre[i]=p+1;
		    else
			    pre[i]=-1;
	    }
    }
};