leetcode第27题--Implement strStr()
Implement strStr().
Returns a pointer to the first occurrence of needle in haystack, or null if needle is not part of haystack.
就是判断haystack中是否有needle,如果包含的话,返回第一次包含的地址。如果没有则返回NULL。
这题官网打出来的是easy,但是要做好绝不简单。我能想到的就是O(m*n)的复杂度了。最经典的是用KMP算法。
class Solution { public: char *strStr(char *haystack, char *needle) { // Start typing your C/C++ solution below // DO NOT write int main() function int lena = strlen(haystack); int lenb = strlen(needle); if(lena < lenb) return NULL; if(lena == lenb) { if(strcmp(haystack, needle)==0) return haystack; return NULL; } for(int i=0; i<=lena-lenb; i++) { bool flag = true; for(int j=0; j<lenb; j++) { if(haystack[i+j] != needle[j]) { flag = false; break; } } if(flag) return haystack + i; } return NULL; } };