6-1 查找子串(BF算法)

问题描述：

C语言标准函数库中包括 strstr 函数，在主串中查找子串。作为练习，我们自己编写一个功能与之相同的函数。

函数原型

char* StrStr(const char *txt, const char *pat);

说明：txt 和 pat 分别为主串和子串的起始地址。若查找成功，则函数值为子串在主串中首次出现的起始地址，否则函数值为NULL。

特别地，我们对C语言库函数strstr进行适当修改：若子串为空串，则没有意义，函数值规定为NULL。

裁判程序

#include <stdio.h> #include <string.h> char* StrStr(const char *txt, const char *pat); int main() { char m[1024], s[1024], *p; gets(m); gets(s); p = StrStr(m, s); if (p) { printf("%d\n", p - m); } else { puts("NULL"); } return 0; } /* 你提交的代码将被嵌在这里 */

输入样例1

This is a pencil
is

输出样例1

2

输入样例2

This is a pencil
be

输出样例2

NULL

代码思路：

1. 首先检查子字符串（pat）是否为空字符串。如果子字符串为空，则无法进行查找，函数直接返回 NULL。
2. 如果子字符串非空，则进入循环，开始查找子字符串在主字符串中的位置。这个循环会一直执行，直到源字符串（txt）遍历完。
3. 在每次循环中，通过两个指针 t 和 p 分别遍历源字符串和子字符串。这两个指针会同时移动，并且比较当前指向的字符是否相等。
4. 如果在某一时刻，*t 和 *p 不相等，或者已经到达子字符串的末尾（即 *p == '\0'），则跳出内层循环。
5. 如果在内层循环结束后，*p == '\0'，这意味着子字符串在源字符串中找到了一个匹配，函数返回当前的 txt 指针，这个指针指向源字符串中子字符串的首次出现位置。
6. 如果遍历完整个源字符串都没有找到匹配的子字符串，函数返回 NULL。

代码：

char* StrStr(const char *txt, const char *pat) {
    if (*pat == '\0') {  // 子串为空字符串，无法查找，返回 NULL
        return NULL;
    }

    while (*txt != '\0') {
        const char *t = txt;
        const char *p = pat;

        while (*t == *p && *p != '\0') {
            t++; p++;
        }

        if (*p == '\0') {  // 匹配成功
            return (char*)txt;
        }

        txt++;
    }

    return NULL;  // 没有匹配成功，返回 NULL
}

运行截图：