KMP算法是字符串模式匹配當中最經典的算法,原來大二學數據結構的有講,但是當時只是記住了原理,但不知道代碼實現,今天終於是完成了KMP的代碼實現。
原理
KMP的原理其實很簡單,給定一個字符串和一個模式串,然後找模式串在給定字符串中的位置。將兩個字符串轉換為字符數組,然後從兩個數組的開始位置"i","j"開始匹配,如果相同,執行"i++","j++"接著比較下一位;如果不相同,就轉到模式串對應next數組的對應位置"next[j]"然後從該位置開始繼續與給定字符串的當前位置"i"進行比較,換句話說就是將模式串提前了"j-next[j]"位繼續比較,不至於每次出現不匹配就又重新回到開始位置進行匹配,充分利用了已匹配過的位置。
代碼
KMP算法的關鍵是得到模式串的next數組:
public static int[] next(char[] p) {
int len = p.length;
int[] next = new int[len];
next[0] = 0;
next[1] = 0; //首先給next[0]和next[1]賦值,這兩個數字是固定的
for(int i = 2; i < len; i++) {
int k = next[i - 1]; //用一個整型數字進行遍歷
while(k >= 0) {
if(p[i - 1] == p[k]) {
next[i] = k + 1; //當匹配到字符時就能得到當前位置的next值,然後結束循環
break;
}
k--;
}
}
return next;
}
得到next數組之後就可以進行KMP匹配:
public int kmpSearch(char[] s, char[] p) {
int i = 0, j = 0; //從0開始
int slen = s.length, plen = p.length;
int[] next = next(p);
while(i < slen && j < plen) {
if(s[i] == p[j]) { //挨個進行匹配
i++;
j++;
} else {
j = next[j]; //如果不相等,返回next[j]位置繼續向後匹配,不用和前面的進行比較
}
}
if(j == plen) //如果匹配到最後,說明匹配成功,返回匹配成功的開始位置
return i - j;
return -1; //否則就是匹配失敗,返回-1
}
KMP算法還有一個進階的next算法,求nextval數組:
public int[] nextVal(char[] p) {
int len = p.length;
int[] nextval = new int[len];
nextval[0] = -1;
int i=-1, j = 0;
while(j < len - 1) {
if(i == -1 || p[j] == p[i]) {
++i;
++j;
if(p[j] != p[i])
nextval[j] = i;
else
nextval[j] = nextval[i];
}else
i = nextval[i];
}
return nextval;
}