返回在字符串1中字符串2首次出现的地址,重新复现strstr(使用KMP算法)

  之前面试碰到一道这样的题目,实现一个函数,返回在字符串1中字符串2首次出现的地址,并打印出处于字符串1的第几位。我当场就想到了用strstr函数,于是很快就写出了下面的代码。

#include 
#include 
int main()
{
   char a[10]="abcdefg";
   char b[10]="cdef";
   char *c;
   int d;
   c=strstr(a,b);
   d=c-a;
   printf("%s,%d",c,d);
   return 0;
}
输出为:
cdefg,2

   但是面试官又说了,不能用strstr函数,没办法,只好重新实现strstr函数了。但现场手撕代码很紧张,只能想出暴力法,于是就有了下面的代码。

#include 
#include 
int my_strstr(const char *str1,const char *str2)
{
	if((str1==NULL||str2==NULL)&&(strlen(str1)

  后来面试完才想到用kmp算法,网上有很多关于kmp算法的介绍,我个人觉得这篇很不错,用的是C++语言实现的。https://blog.csdn.net/starstar1992/article/details/54913261/

代码如下,只是改了一些地方,用C语言实现出来。

#include 
#include 
#include 
void cal_next(char *str, int *next, int len)
{
    next[0] = -1;//next[0]初始化为-1,-1表示不存在相同的最大前缀和最大后缀
    int k = -1;//k初始化为-1
    for (int q = 1; q <= len-1; q++)
    {
        while (k > -1 && str[k + 1] != str[q])//如果下一个不同,那么k就变成next[k],注意next[k]是小于k的,无论k取任何值。
        {
            k = next[k];//往前回溯
        }
        if (str[k + 1] == str[q])//如果相同,k++
        {
            k = k + 1;
        }
        next[q] = k;//这个是把算的k的值(就是相同的最大前缀和最大后缀长)赋给next[q]
    }
}
int KMP(char *str, int slen, char *ptr, int plen)
{
    int * next = (int *)malloc(plen*sizeof(int));
    cal_next(ptr, next, plen);//计算next数组
    int k = -1;
    for (int i = 0; i < slen; i++)
    {
        while (k >-1&& ptr[k + 1] != str[i])//ptr和str不匹配,且k>-1(表示ptr和str有部分匹配)
            k = next[k];//往前回溯
        if (ptr[k + 1] == str[i])
            k = k + 1;
        if (k == plen-1)//说明k移动到ptr的最末端
        {
            //cout << "在位置" << i-plen+1<< endl;
            //k = -1;//重新初始化,寻找下一个
            //i = i - plen + 1;//i定位到该位置,外层for循环i++可以继续找下一个(这里默认存在两个匹配字符串可以部分重叠),感谢评论中同学指出错误。
            return i-plen+1;//返回相应的位置
        }
    }
    return -1;  
}
int main()
{
   char a[10]="abcdefg";
   char b[10]="cdef";
   int d;
   d=KMP(a,strlen(a),b,strlen(b));
   printf("%d",d);
   return 0;
}
输出为:
2

 

你可能感兴趣的:(C/C++面试题)