字符串问题python实现（三）

问题：写一个函数模拟C++中strstr()函数：该函数的返回值是主串中字符子串的位置以后的所有字符，比如主串       是"123456"，子串是"234"，那么函数的返回值就是"2345678"。

python版代码：

# -*- coding:utf-8 -*-
# 问题：写一个函数模拟C++中strstr()函数：该函数的返回值是主串中字符子串的位置以后的所有字符，
#      比如主串是"123456"，子串是"234"，那么函数的返回值就是"2345678"。 
# by chasdmeng

def strstr(string, strCharSet):
    for i in range(len(string)):
        temp = i
        j = 0
        if string[i] == strCharSet[j]:
            while string[i] == strCharSet[j]:
                i+=1
                j+=1
                if j == len(strCharSet)-1:
                    return string[i-j:]
            i = temp
    return None

if __name__ == '__main__':
    string = "12345554555123"
    strCharSet = "234"

    print strstr(string, strCharSet)