LeetCode刷题笔记三

28. Implement strStr()

Description：

Implement strStr().

Returns the index of the first occurrence of needle in haystack, or -1 if needle is not part of haystack.

算法：KMP算法

参考资料：http://blog.csdn.net/yutianzuijin/article/details/11954939/

代码：

class Solution {
public:
    int strStr(string haystack, string needle) 
    {
        int len1=haystack.size(), len2=needle.size();
        if(len1 < len2)
        {
            return -1;
        }
        if(len1==0 && len2==0)
        {
            return 0;
        }
        if(len2==0)
        {
            return 0;
        }
        int i, j;
        int *next = new int[len2+1]; //next[i]表示needle前i个元素中前缀和后缀的最大公共长度
        next[0] = 0;
        next[1] = 0;
        for(i=1; i<len2; ++i)
        {
            j = next[i];
            while(j>0 && needle[i] != needle[j])
            {
                j=next[j];
            }
            if(needle[i] == needle[j])
            {
                ++j;
            }
            next[i+1] = j;
        }
        for(i=0, j=0; i<len1; ++i)
        {
            while(j>0 && haystack[i] != needle[j])
            {
                j = next[j];
            }
            if(haystack[i] == needle[j])
            {
                j += 1;
            }
            if(j == len2)
            {
                delete [] next;
                return i-j+1;
            }
        }
        // for(i=0, j=0; i<=len1-len2;)
        // {
        //     while(j<len2 && haystack[i+j] == needle[j])
        //     {
        //         ++j;
        //     }
        //     if(j == len2)
        //     {
        //         delete [] next;
        //         return i;
        //     }
        //     else
        //     {
        //         i += j-next[j];
        //         j = 0;
        //     }
        // }
        delete [] next;
        return -1;
    }
};