自写strstr函数

给出一个字符串和另一个字符串，若后者为前者的子字符串，则返回其所在位置的起始指针。

#include<iostream>
#include<string>
#include<stdio.h>
using namespace std;
const char* strs(const char*, const char*);
int main()
{
    char *str=new char[20];
    char *substr=new char[20];
    gets(str);
    gets(substr);
    cout<<strs(str,substr);
    return 0;
}

const char* strs(const char* str, const char* substr)
{
    int j=0;
    string str1=str;
    int n=str1.length();
    for (int i=0;i<n;i++)
    {
        j=0;
        while(str[i+j]==substr[j])
        {
            if(substr[j+1]=='\0')
                return str+i;
                j++;
        }
    }
    return NULL;
}