在刷代码随想录的时候,遇到了leetcode这道经典题目
28. 实现 strStr()
力扣题目链接(opens new window)
实现 strStr() 函数。
给定一个 haystack 字符串和一个 needle 字符串,在 haystack 字符串中找出 needle 字符串出现的第一个位置 (从0开始)。如果不存在,则返回 -1。
示例 1: 输入: haystack = "hello", needle = "ll" 输出: 2
示例 2: 输入: haystack = "aaaaa", needle = "bba" 输出: -1
说明: 当 needle 是空字符串时,我们应当返回什么值呢?这是一个在面试中很好的问题。 对于本题而言,当 needle 是空字符串时我们应当返回 0 。这与C语言的 strstr() 以及 Java的 indexOf() 定义相符。
当然,首先我们根据模式串来写出,相对应的前缀表 例如 a a b a a f 这个模式串就是如下的前缀表 定义了 i j i j [0 1 0 1 2 0]
如何去求这个next数组呢
void getnext(int*next,const string&s){next[0]=0;int j=0;for(int i=1;i<s.size();i++){while(j>0&&s[i]!=s[j]){j=next[j-1]; }if(s[i]==s[j]){++j;}next[i]=j;}}
i 和 j 不相等就回退
最后的话,就是匹配!
整体代码如下
int strStr(string haystack, string needle) {if(needle.size()==0){return 0;}int next[needle.size()];getnext(next,needle);int j=0;for(int i=0;i<haystack.size();i++){ while(j>0&&haystack[i]!=needle[j]){j=next[j-1];}if(haystack[i]==needle[j]){j++;}if(j==needle.size()){return (i - j + 1);//i是文本串的总长索引-模式的总索引+1}}return -1;}
i用来遍历 文本串,j 用来遍历模式串,同样,就是相等++,不同的话,这时候 j就是回退next[j-1],此时刚好就是 最长的相等的前缀长度,如果 以 aabaabaaf aabaaf 这两个为例,那么就会退到b 然后重新匹配