C++实现字符串匹配strstr()函数

//函数接口：char* mystrstr(char* str1,char* str2)
//解释：
//在字符串str1中查找第一次出现str2的位置
//如果找到匹配的字符串，返回第一次匹配字符串的指针
//否则，返回NULL

#include<isotream>

using namespace std;

char* mystrstr(char* str1,char* str2)
{
	char* src,*sub;

	if(str1==NULL || str2==NULL)
	{
		cout<<"ERROR!"<<endl;
		return NULL;
	}

	while(*str1!='\0')
	{
		src=str1;
		sub=str2;
		do
		{
			if(*sub=='\0')
			{
				return str1;//找到子串
			}
		}while(*src++==*sub++);
		str1++;
	}
	return NULL;
}

int main()
{
	char str1[]="ABCD1234";
	char str2[]="1234";
	cout<<mystrstr(str1,str2)<<endl;
	return 0;
}