问题描述:
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
代码思路:
- 首先检查子字符串(pat)是否为空字符串。如果子字符串为空,则无法进行查找,函数直接返回 NULL。
- 如果子字符串非空,则进入循环,开始查找子字符串在主字符串中的位置。这个循环会一直执行,直到源字符串(txt)遍历完。
- 在每次循环中,通过两个指针 t 和 p 分别遍历源字符串和子字符串。这两个指针会同时移动,并且比较当前指向的字符是否相等。
- 如果在某一时刻,*t 和 *p 不相等,或者已经到达子字符串的末尾(即 *p == '\0'),则跳出内层循环。
- 如果在内层循环结束后,*p == '\0',这意味着子字符串在源字符串中找到了一个匹配,函数返回当前的 txt 指针,这个指针指向源字符串中子字符串的首次出现位置。
- 如果遍历完整个源字符串都没有找到匹配的子字符串,函数返回 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
}
运行截图: