一天一道LeetCode系列
(一)题目
Implement strStr().
Returns the index of the first occurrence of needle in haystack, or -1 if needle is not part of haystack.
(二)解题
第一种解法:朴素匹配算法
/*两个指针,分别指向两个字符串的首字符如果相等则一起向后移动,如果不同i取第一个相同字符的下一个开始继续匹配如果最后j等于needle的长度则匹配成功,返回i-j否则返回0*/class Solution {public: int strStr(string haystack, string needle) { int j,i; for(i = 0 , j =0 ; i<haystack.length() && j < needle.length() ;) { if(i+needle.length()>haystack.length()) return -1; if(haystack[i]==needle[j]){//如果匹配上就继续向后匹配 i++; j++; } else{ i-=j-1;//回溯到匹配开始时needle的首字符对应的下一位 j=0;//j回溯到needle的首字符 } } if(j==needle.length()) return i-j; else return -1; }};
第二种解法:KMP模式匹配算法
关于kmp,请自行百度或者大话数据结构P143页
class Solution {public: int strStr(string haystack, string needle) { int hlen = haystack.length(); int nlen = needle.length(); if(hlen==0) return nlen==0?0:-1;//临界值判断 if(nlen==0) return 0;//needle为NULL,就直接返回0 int* next = new int[nlen+1]; getNext(needle,next); int i = 0; int j = 0; while(i<hlen&&j<nlen){ if(j==-1 || haystack[i]==needle[j]){ i++;j++; } else j=next[j]; } if(j==nlen) return i-j;//等于nlen代表匹配成功,返回i-j即needle首字符在haystack中的位置 else return -1; } void getNext(string& needle,int next[]) { int i = 0; int j = -1; next[0] = -1; while(i<needle.length()){ if(j==-1 || needle[i]==needle[j]){ i++; j++; if(needle[i] == needle[j]) next[i] = next[j];//kmp优化,防止aaaaab和aaaac前四位的无效 else next[i] = j; } else j=next[j]; } }};