leetcode 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了。关于kmp算法就不多说了，是很经典的问题了，我只想补充一下，kmp有两种写法，有一个优化版本的。这两个版本不同之处在于求next数组的不同方法，我使用的是一般的next数组，next数组
就是保存最长前缀和后缀的匹配重度。
如果不理解kmp，搜下到处都有了，好久不看，自己推next推了很久才想起来。

class Solution {
public:

void get_n(const char * pattern,int *p,int n)
{
p[0]=-1;
for(int i=1;i<n;i++)
{
int j=p[i-1];
while(j>=0&&pattern[j+1]!=pattern[i])
j=p[j];
if(pattern[j+1]==pattern[i])
p[i]=j+1;
else
p[i]=-1;
}
}
char *strStr(char *haystack, char *needle) {
if(!haystack||!needle)
return NULL;
int m=0,n=0;
while(haystack[m])m++;
while(needle
)n++;
if(m<n)
return NULL;
int next
;
get_n(needle,next,n);
int i=0,j=0;
while(i<m&&j<n)
{
if(haystack[i]==needle[j])
{
i++;j++;
}
else
if(j==0)
i++;
else
{
j=next[j-1]+1;
}
}
if(j==n)
return haystack+i-j;
return NULL;
}
};
如有不正之处请指正。