程序员面试宝典：模拟strstr()函数

請写一个函数来模拟c++中的strstr()函数：该函数的返回值是主串中子串及以后的字符全部返回

#include<iostream>
#include<string>
#include<cstring>
#include<cstdlib>
using namespace std;

const char* str_str(const char* str_src,const char* str_set)
{
	int len1=strlen(str_src);
	int len2=strlen(str_set);
	int first;
	for(int i=0;i<len1;i++)
	{
		first=i;
		int k=0;
		while(str_src[k+i]==str_set[k] && k<len2)
			k++;
		if(k==len2)   return &str_src[first];		
	}
	return NULL;
}

int main()
{
	cout<<"Input the original string:";
	string str_src;
	cin>>str_src;
	cout<<"Input the substring:";
	string str_sub;
	cin>>str_sub;
	char *str1,*str2;
	str1=new char[str_src.length()+1];
	str2=new char[str_sub.length()+1];
	strcpy(str1,str_src.c_str());
	strcpy(str2,str_sub.c_str());
	cout<<str_str(str1,str2)<<endl;
	delete str1;
	delete str2;
	return 0;
}