编程实现strstr函数

函数接口为：char *mystrstr(char *str1,char *str2)

要求：在字符串str1中查找第一次出现字符串str2的位置，如果找到匹配的字符串，返回第一次匹配的指针，否则返回NULL。

#include<iostream>
using namespace std;
char *mystrstr(char *str1, char *str2)
{
    char *p, *q;
    if (str1 == NULL || str2 == NULL)
   {
    cout << "the string is error!" <<endl;
    exit(0);
   }
  while (*str1 != '\0')
  {
    p = str1;
    q = str2;
  while (*p == *q&&*q!='\0')
  {
    p++;
    q++;
    }
    if (*q == '\0')
      return str1;
    str1++;
  }
  return NULL;
}
int main()
{
  char a[] = "hello world";
  char b[] = "world";
  char *p = mystrstr(a, b);
  if(p)
  cout << p << endl;
  getchar();
  return 0;
}