题目:
Implement strStr().
Returns the index of the first occurrence of needle in haystack, or -1 if needle is not part of haystack.
翻译:实现一个方法strStr()。返回字符串needle第一次在字符串haystack出现的下标,如果needle不是haystack的一部分,就返回-1。
分析:
在文本中查找某个模式出现的位置的算法,称为字符串匹配算法。常用的方法有朴素字符串匹配算法、KMP算法等。朴素字符串匹配算法,就是把2个字符串头部对齐,然后逐一字符匹配,失配后,把needle右移一位,继续从头匹配。
Java版本:
class Solution {
public int strStr(String haystack, String needle) {
/**brute force的算法,假设原串的长度是n,匹配串的长度是m。思路很简单,
就是对原串的每一个长度为m的字串都判断是否跟匹配串一致。总共有n-m+1个子串,
所以算法时间复杂度为O((n-m+1)*m)=O(n*m),空间复杂度是O(1)。**/
// if(haystack==null || needle == null || needle.length()==0)
// return 0;
// if(needle.length()>haystack.length())
// return -1;
// for(int i=0;i<=haystack.length()-needle.length();i++)
// {
// boolean successFlag = true;
// for(int j=0;j<needle.length();j++)
// {
// if(haystack.charAt(i+j)!=needle.charAt(j))
// {
// successFlag = false;
// break;
// }
// }
// if(successFlag)
// return i;
// }
// return -1;
// }
// 另一种简洁算法:
int l1 = haystack.length(), l2 = needle.length();
if (l1 < l2) {
return -1;
} else if (l2 == 0) {
return 0;
}
int threshold = l1 - l2;
for (int i = 0; i <= threshold; ++i) {
if (haystack.substring(i,i+l2).equals(needle)) {
return i;
}
}
return -1;
}
}
注意的是,从haystack第一个元素下标0开始,界限为两个数组的差值threshold,从haystack第一个元素+needle的长度与needle进行判断比较,类似个方块形状来进行块状的匹配一样。
Python版本:
class Solution(object):
def strStr(self, haystack, needle):
"""
:type haystack: str
:type needle: str
:rtype: int
"""
for i in range(len(haystack) - len(needle)+1):
if haystack[i:i+len(needle)] == needle:
return i
return -1
class Solution {
public:
int strStr(string haystack, string needle) {
int m = haystack.length(), n = needle.length();
if (!n) return 0;
for (int i = 0; i < m - n + 1; i++) {
int j = 0;
for (; j < n; j++)
if (haystack[i + j] != needle[j])
break;
if (j == n) return i;
}
return -1;
}
};