【leetcode】Implement strStr()

Implement strStr()

Implement strStr().

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

 

最简单的思路，逐一比较：

class Solution {
public:
    int strStr(char *haystack, char *needle) {
       
        int n1=strlen(haystack);
        int n2=strlen(needle);          
        int result=-1;
        bool flag;
        for(int i=0;i<n1-n2+1;i++)
        {
            flag=true;
            for(int j=0;j<n2;j++)
            {
                if(haystack[i+j]==needle[j])
                {
                    continue;
                }
                else
                {
                    flag=false;
                    break;
                }
            }
           
            if(flag==true)
            {
                result=i;
                break;
            }
        }          
        return result;
    }
};

 

可以采用KMP算法：

KMP算法的核心是，当比较某两个字符不相等时，不必从头开始重新比较，而是根据引入的next，确定子串回溯的位置。

 

 

举个例子：

 

一个子串：

a	b	a	a	b	a	b	a
-1	0	0	1	1	2	3	2

 

next[j]表示了如果在比较字符串时，needle[j]!=haystack[k]

那么下一次比较时，从next[j]所指定的位置进行比较，即j=next[j],k不变

 

 

 

 

 

class Solution {
public:
    int strStr(char *haystack, char *needle) {         
       
        int i=0;
        int j=0;
        int n1=strlen(haystack);
        int n2=strlen(needle);
         
        vector<int> next=getNext(needle);
       
        while(i<n1&&j<n2)
        {
            if(j==-1||haystack[i]==needle[j])
            {
                i++;
                j++;
            }
            else
            {
                j=next[j];
            }
        }
       
        if(j==n2)
        {
            return i-j;
        }
        else
        {
            return -1;
        }
    }
   
    vector<int> getNext(char *needle)
    {
        int n=strlen(needle);
       
        vector<int> next(n);
        if(n==0)
        {
            return next;
        }
       
        next[0]=-1;
       
        int i=0;
        int k=-1;
        while(i<n-1)
        {
            if(k==-1||needle[k]==needle[i])
            {
                
                i++;
                k++;
                next[i]=k;
            }
            else
            {               
                k=next[k];
            }
        }
       
        return next;
       
    }
 
};

 

 对于k=next[k]

举个例子：如果比较最

a	b	a	a	b	a	b	a
-1	0	0	1	1	2	3	2

 

最后一个元素的next为2

这是因为，倒数第二个元素与第四个元素不相等，此时

k=next[3]=1;

 

继续循环，

b==needle[k]=needle[1]

 

所以

i++

k++

得到

next[7]=2;