用这题复习下kmp算法。kmp网上坑爹的是有很多种匹配方式,容易混淆,后人要警惕啊。这里对要查找的字符串的next全部置为-1,预处理时i和next[i-1]+1相比较。和http://kb.cnblogs.com/page/176818/ 这里相似。预处理完再匹配,第i个字符不匹配则比较第next[i-1]+1个。
class Solution{ public: char *strStr(char *haystack, char *needle){ if(haystack == NULL || needle == NULL) return NULL; int sz = strlen(needle); vector<int>next(sz,-1); for(int i = 1; i < sz; i++){ int index = i - 1; while(index != -1 && needle[next[index]+1] != needle[i]) index = next[index]; next[i] = index==-1?-1:next[index]+1; } int i = 0; while(*haystack && i <sz){ while(i != -1 && *haystack != needle[i]) i = i==0?-1:next[i-1]+1; haystack++; i++; } return i < sz?NULL:haystack-sz; } }; /* a b C b C a d -1 -1 -1 -1 -1 0 -1 abcab cabdabba a b c a b d -1 -1 -1 0 1 -1*/
时间: 2024-10-25 08:58:11