实现 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() 定义相符。
总结:
还没完全弄懂KMP算法,先把代码附上。。
- (简单匹配)
顾名思义,简单暴力,二层循环,一层遍历文本串(注意,只需要遍历文本串减模式串的长度),一层模式串,如果模式串遍历完了都没有不相同的元素,便可返回此刻文本串当前下标的位置。
//简单匹配
class Solution {
public:
//匹配某位置开始的
bool match(string h,string n,int h_start,int n_start,int &h_fail,int &n_fail)
{
int i=h_start,j=n_start;
for(;j<n.size();i++,j++)
{
if(h[i]!=n[j])
{
h_fail=i;
n_fail=j;
return false;
}
}
return true;
}
int strStr(string haystack, string needle) {
int h_start=0,n_start=0,h_fail,n_fail;
if(needle!="")
{
if(haystack=="")return -1;
}
else return 0;
//1简单匹配
for(h_start=0;h_start<=haystack.size()-needle.size();h_start++)
{
if(match(haystack,needle,h_start,n_start,h_fail,n_fail))
return h_start;
}
return -1;
}
};
//kmp算法1
class Solution {
public:
//匹配某位置开始的
bool match(string h,string n,int h_start,int n_start,int &h_fail,int &n_fail)
{
int i=h_start,j=n_start;
for(;j<n.size();i++,j++)
{
if(h[i]!=n[j])
{
h_fail=i;
n_fail=j;
return false;
}
}
return true;
}
//计算偏移函数
void findPi(string p,vector<int>&pi)
{
pi[0]=0;
int m=p.size();
int k=0;
for(int q=1;q<m;q++)
{
while(k>0 && p[k]!=p[q])
k=pi[k-1];
if(p[k]==p[q])k++;
pi[q]=k;
}
}
int strStr(string haystack, string needle)
{
vector<int> pi(needle.size()+1,0);
int h_start=0,n_start=0,h_fail,n_fail;
if(needle!="")
{
if(haystack=="")return -1;
}
else return 0;
//KMP 快速匹配
findPi(needle,pi);
for(int i=0,q=0;i<haystack.size();i++)
{
while(q>0 && haystack[i]!=needle[q])
q=pi[q-1];
if(haystack[i]==needle[q])q++;
if(q==needle.size())return i-needle.size()+1;
}
return -1;
}
};
//leetcode最佳范例
class Solution {
public:
int strStr(string haystack, string needle) {
unsigned length = needle.size();
if (haystack.size() < length)return -1;
if (length == 0) return 0;
for (unsigned i = 0; i < haystack.size() - length + 1; i++)
{
if (haystack.substr(i, length) == needle)return i;
}
return -1;
}
};
//KMP算法2
class Solution {
public:
int strStr(string haystack, string needle) {
int size1 = haystack.size(), size2 = needle.size(), i = 0, j = 0;
if (size2 == 0) {
return 0;
}
int* next = new int[size2];
getNext(needle, next);
while (i < size1) {
if (j == -1 || haystack[i] == needle[j])
{
i++;
j++;
}
else
{
j = next[j];
}
if (j == size2)
{
return i - size2;
}
}
return -1;
}
void getNext(string s, int* next) {
int j = 0, k = -1;
next[0] = -1;
while (j < s.size() - 1) {
if (k == -1 || s[j] == s[k]) {
j++;
k++;
next[j] = k;
}
else {
k = next[k];
}
}
}
};