怎么写自己的strstr函数

在C语言编程中，strstr函数是一个非常有用的字符串处理工具，它可以搜索一个字符串中是否包含另一个子串，并返回子串的指针。但在某些情况下，我们可能需要编写自己的strstr函数，以满足特定的需求。本文将详细讲解如何编写自己的strstr函数。 首先，让我们总结一下strstr函数的功能。strstr函数接收两个参数：一个是源字符串，另一个是要查找的子串。如果找到子串，则返回子串第一次出现的指针；如果没有找到，则返回NULL。 以下是编写自己的strstr函数的步骤：

1. 验证输入参数的有效性。确保源字符串和子串都不是空指针。
2. 如果子串为空，根据strstr的定义，应返回源字符串的指针。
3. 遍历源字符串，使用双重循环进行子串匹配。
4. 在内部循环中，比较当前源字符串位置的字符与子串中的字符是否相等。如果所有字符都匹配，则找到子串，返回当前源字符串的指针。
5. 如果遍历完源字符串后仍没有找到子串，返回NULL。 下面是一个自定义strstr函数的示例代码：

char *my_strstr(const char *haystack, const char *needle) {
    if (needle == NULL) return (char *)haystack;
    if (*needle == '\0') return NULL;
    while (*haystack) {
        const char *h = haystack;
        const char *n = needle;
        while (*n && (*h == *n)) {
            h++;
            n++;
        }
        if (*n == '\0') return (char *)haystack;
        haystack++;
    }
    return NULL;
}

最后，我们来总结一下。编写自己的strstr函数需要遵循一些基本步骤，包括验证输入参数、遍历源字符串、匹配子串等。通过实现自己的函数，我们可以更深入地了解字符串处理，并在必要时对性能进行优化。