strstr函数模拟实现（c语言）

首先，我们来看strstr函数的使用

#include<stdio.h>
#include<string.h>
int main()
{
	char arr1[] = "abccdef";
	char arr2[] = "cde";

	char* ret = strstr(arr1, arr2);
	if (ret == NULL)
	{
		printf("子串不存在\n");
	}
	else
		printf("%s\n", ret);
	return 0;
}

可见，strstr函数是用来查找 字串的一个函数。

因为字符串中有“cde"子串，所以代码运行起来结果就是”cdef" 。

现在，我们已经将strstr函数的作用大概了解了一下，下面开始函数的模拟实现。

---

首先，在cplusplus中搜索strstr可以了解到她返回的类型及参数

 char * strstr ( char * str1, const char * str2 );

相关链接： strstr - C++ Reference (cplusplus.com)

---

思路

接下来，我们分析实现思路

 

1.用while循环将*s1与*s2进行比较，由图可见，*s1开始并不相等，所以p++，再将p赋给s1.

char* my_strstr(char* str1, const char* str2)
{
	const char* s1 = str1;
	const char* s2 = str2;
	char* p = str1;
	while (*p)
	{
		s1 = p;
		while (*s1 == *s2 )
		{
			;
		}
		p++;
	}
}

直到s1到了下图的位置，两者相等

 

 2.判断是否“cde"在”ccdef“中

当*s1==*s2时进入while循环，s1++,s2++.逐个比较。

当然，用膝盖想也知道（doge.）由于我选择的示例比较特殊，第一次进入循环时，并不能直接判断出来是否“cde"在”ccdef“中，所以跳出循环后p++，s1 = p.进入第二次循环

并且，s2每次循环后，需要回到原来的位置。所以s2 = str2.

char* my_strstr(char* str1, const char* str2)
{
	const char* s1 = str1;
	const char* s2 = str2;
	char* p = str1;
	while (*p)
	{
		s1 = p;
		s2 = str2;
		while (*s1 == *s2)
		{
			s1++;
			s2++;
		}
		p++;
	}
}

现在主体已经实现了，现在只剩下一个问题，就是当s1或者s2指向'\0'该怎么办。

当s1指向'\0'时，p也就指向了'\0'，代表字符串已经被找完了，但是没有找到子字符串"cde"这时，循环while(*p)结束，返回空指针。

当s2指向’\0'时，表示在”abccdef"中找到了"cde“，那么就返回p指向的地址。

char* my_strstr(char* str1, const char* str2)
{
	const char* s1 = str1;
	const char* s2 = str2;
	char* p = str1;
	while (*p)
	{
		s1 = p;
		s2 = str2;
		while (*s1 == *s2 && *s1 != '\0' && *s2 != '\0')
		{
			s1++;
			s2++;
		}
		if (*s2 == '\0')
		{
			return p;
		}
		p++;
	}
	return NULL;
}

最后再进行部分优化。进行断言

#include<stdio.h>
#include<assert.h>

char* my_strstr(char* str1, const char* str2)
{
	assert(str1 && str2);
	const char* s1 = str1;
	const char* s2 = str2;
	char* p = str1;
	while (*p)
	{
		s1 = p;
		s2 = str2;
		while (*s1 == *s2 && *s1 != '\0' && *s2 != '\0')
		{
			s1++;
			s2++;
		}
		if (*s2 == '\0')
		{
			return p;
		}
		p++;
	}
	return NULL;
}
int main()
{
	char arr1[] = "abccdef";
	char arr2[] = "cde";

	char* ret = my_strstr(arr1, arr2);
	if (ret == NULL)
	{
		printf("子串不存在\n");
	}
	else
		printf("%s\n", ret);
	return 0;
}

运行结果：

 

结束