给你两个字符串 haystack 和 needle ,请你在 haystack 字符串中找出 needle 字符串出现的第一个位置(下标从 0 开始)。如果不存在,则返回 -1 。
我的思路:这题比较简单,直接遍历haystack找needle即可
代码如下:
class Solution {
public int strStr(String haystack, String needle) {
if (haystack.isEmpty() && needle.isEmpty()){
return 0;
}
int length = needle.length();
for (int i = 0; i < haystack.length() - length + 1; i++) {
if (haystack.substring(i, i+length).equals(needle)){
return i;
}
}
return -1;
}
}
官方采用了KMP算法,比较复杂,感觉用在这题算是大题小做了.
将代码实现分为两部分:
第一部分是求 needle 部分的前缀函数,我们需要保留这部分的前缀函数值。
第二部分是求haystack 部分的前缀函数,我们无需保留这部分的前缀函数值,只需要用一个变量记录上一个位置的前缀函数值即可。当某个位置的前缀函数值等于 m 时,说明我们就找到了一次字符串needle 在字符串haystack 中的出现(因为此时真前缀恰为字符串needle,真后缀为以当前位置为结束位置的字符串haystack 的子串),我们计算出起始位置,将其返回即可。
class Solution {
public int strStr(String haystack, String needle) {
int n = haystack.length(), m = needle.length();
if (m == 0) {
return 0;
}
int[] pi = new int[m];
for (int i = 1, j = 0; i < m; i++) {
while (j > 0 && needle.charAt(i) != needle.charAt(j)) {
j = pi[j - 1];
}
if (needle.charAt(i) == needle.charAt(j)) {
j++;
}
pi[i] = j;
}
for (int i = 0, j = 0; i < n; i++) {
while (j > 0 && haystack.charAt(i) != needle.charAt(j)) {
j = pi[j - 1];
}
if (haystack.charAt(i) == needle.charAt(j)) {
j++;
}
if (j == m) {
return i - m + 1;
}
}
return -1;
}
}