题目:
实现 strStr() 函数。
给定一个 haystack 字符串和一个 needle 字符串,在 haystack 字符串中找出 needle 字符串出现的第一个位置 (从0开始)。如果不存在,则返回 -1。
示例 1:
输入: haystack = "hello", needle = "ll"
输出: 2
示例 2:
输入: haystack = "aaaaa", needle = "bba"
输出: -1
说明:
当 needle 是空字符串时,我们应当返回什么值呢?这是一个在面试中很好的问题。
对于本题而言,当 needle 是空字符串时我们应当返回 0 。这与C语言的 strstr() 以及 Java的 indexOf() 定义相符。
解答:
int strStr(char* haystack, char* needle) {
int len=strlen(haystack),sublen=strlen(needle);
if(sublen>len)
return -1;
if(sublen==0)
return 0;
if(sublen==1)
{
for(int i=0;i<len;i++)
{
if(haystack[i]==needle[0])
return i;
}
return -1;
}
char *a=(char *)malloc(sizeof(char)*sublen);
a[0]=0;
int i,j;
for(i=1,j=0;i<sublen;)
{
if(needle[i]==needle[j])
{
a[i]=j+1;
i++;
j++;
continue;
}
if(needle[i]!=needle[j])
{
if(j>0)
{
j--;
continue;
}
else
{
a[i]=0;
i++;
}
}
}
for(i=0,j=0;i<len;)
{
if(j>=sublen)
return i-j;
if(haystack[i]==needle[j])
{
i++;
j++;
continue;
}
if(haystack[i]!=needle[j])
{
if(j>=1)
{
j=a[j-1];
}
else if(j==0)
{
i++;
}
}
}
if(i==len && j==sublen)
return i-j;
return -1;
}
KMP算法