C++ - 实现strstr函数

/*
函数名: strstr
功能：找出str2字符串在str1字符串中第一次出现的位置（不包括str2的串结束符）。
返回值：返回该位置的指针；如找不到，返回空指针。
*/

#include <iostream>

#define NULL 0

using namespace std;

char* myStrstr(const char* s1, const char* s2)
{
	int n;
	if(*s2)
	{
		while(*s1)
		{
			for(n = 0; *(s1 + n) == *(s2 + n); n++)
			{
				if(!*(s2 + n + 1))
				{
					return (char*)s1;
				}
			}

			s1++;
		}

		return NULL;
	}
	else
	{
		return (char*)s1;
	}
}

int main()
{
	char s1[10] = "453456789";
	char s2[6] = "45678";
	char* s = myStrstr(s1, s2);
	cout << s << endl;
	return 0;
}

// Output:
/*
456789
*/