代码先放上,具体实现方法和原理下次补。(咕咕咕。。补了你们也不一定会看)
如果哪里有可以优化的话请告诉我,我改进改进。
func kmp(str1, str2 string) int {
// count next_arr
if len(str1)<len(str2) {
return -1
}
next_arr := make([]int, len(str2))
for i := 0; i < len(str2); i++ {
// 这里一般情况下 i==0时应该是-1,但是为了减少代码,
// 所以合成0,不影响算法的结果
t := 0
for j := 0; j < i-1; j++ {
if str2[:j+1] == str2[i-(j+1):i] {
t = j + 1
}
}
next_arr[i] = t
}
// 初始化i,y,index 这里应该都看得懂
i, y, index := 0, 0, -1
// 这里就是开始kmp算法了,针对自己的习惯,对其进行了小部分改造
for i <= len(str1)-len(str2) {
for j := y; j <= len(str2); j++ {
if j == len(str2) {
index = i
i++
y = 0
break
} else if str1[i+j] != str2[j] {
if next_arr[j] < 1 {
for j > 1 {
if next_arr[j] >= 1 {
i = i + (j - next_arr[j])
y = next_arr[j]
break
}
j--
}
if j <= 1 {
i++
y = 0
}
break
} else {
i = i + (j - next_arr[j])
y = next_arr[j]
break
}
}
}
}
return index
}