字符串kmp算法詳解

之前要研究aho-corasick算法 拖了好久  感覺自己博客要開始了!!


aho-corasick算法依賴2元素:

1.Trie樹解析,1個月前就已經寫過博客分析過了。

2.KMP算法


此文重點介紹字符串KMP算法:


一開始說說普通模式算法("BF"算法)思路:模式串從主串的第一個字符開始匹配,每匹配失敗,主串中記錄匹配進度的指針 i 都要進行 i-j+1 的回退操作(這個過程稱爲“指針回溯”),同時模式串向後移動一個字符的位置。一次次的循環,直到匹配成功或者程序結束。


"KMP"算法相比於"BF"算法,優勢在於:

  • 在保證指針 i 不回溯的前提下,當匹配失敗時,讓模式串向右移動最大的距離;

  • 並且可以在O(n+m)的時間數量級上完成對串的模式匹配操作;


故,"KMP"算法稱爲“快速模式匹配算法”

模式串向右移動距離的計算

在模式串和主串匹配時,各有一個指針指向當前進行匹配的字符(主串中是指針 i ,模式串中是指針 j ),在保證 i 指針不回溯的前提下,如果想實現功能,就只能讓 j 指針回溯。

j 指針回溯的距離,就相當於模式串向右移動的距離。 j 指針回溯的越多,說明模式串向右移動的距離越長。

計算模式串向右移動的距離,就可以轉化成:當某字符匹配失敗後, j 指針回溯的位置。

對於一個給定的模式串,其中每個字符都有可能會遇到匹配失敗,這時對應的 j 指針都需要回溯,具體回溯的位置其實還是由模式串本身來決定的,和主串沒有關係。

模式串中的每個字符所對應 j 指針回溯的位置,可以通過算法得出,得到的結果相應地存儲在一個數組中(默認數組名爲 next )。

計算方法是:對於模式串中的某一字符來說,提取它前面的字符串,分別從字符串的兩端查看連續相同的字符串的個數,在其基礎上 +1 ,結果就是該字符對應的值。

每個模式串的第一個字符對應的值爲 0 ,第二個字符對應的值爲 1 。

例如:求模式串 “abcabac” 的 next 。前兩個字符對應的 0 和 1 是固定的。

對於字符 ‘c’ 來說,提取字符串 “ab” ,‘a’ 和 ‘b’ 不相等,相同的字符串的個數爲 0 ,0 + 1 = 1 ,所以 ‘c’ 對應的 next 值爲 1 ;

第四個字符 ‘a’ ,提取 “abc” ,從首先 ‘a’ 和 ‘c’ 就不相等,相同的個數爲 0 ,0 + 1 = 1 ,所以,‘a’ 對應的 next 值爲 1 ;

第五個字符 ‘b’ ,提取 “abca” ,第一個 ‘a’ 和最後一個 ‘a’ 相同,相同個數爲 1 ,1 + 1 = 2 ,所以,‘b’ 對應的 next 值爲 2 ;

第六個字符 ‘a’ ,提取 “abcab” ,前兩個字符 “ab” 和最後兩個 “ab” 相同,相同個數爲 2 ,2 + 1 = 3 ,所以,‘a’ 對應的 next 值爲 3 ;

最後一個字符 ‘c’ ,提取 “abcaba” ,第一個字符 ‘a’ 和最後一個 ‘a’ 相同,相同個數爲 1 ,1 + 1 = 2 ,所以 ‘c’ 對應的 next 值爲 2 ;

所以,字符串 “abcabac” 對應的 next 數組中的值爲(0,1,1,1,2,3,2)。

上邊求值過程中,每次都需要判斷字符串頭部和尾部相同字符的個數,而在編寫算法實現時,對於某個字符來說,可以借用前一個字符的判斷結果,計算當前字符對應的 next 值。

具體的算法如下:

模式串T爲(下標從1開始):“abcabac”
next數組(下標從1開始):    01

第三個字符 ‘c’ :由於前一個字符 ‘b’ 的 next 值爲 1 ,取 T[1] = ‘a’ 和 ‘b’ 相比較,不相等,繼續;由於 next[1] = 0,結束。 ‘c’ 對應的 next 值爲1;(只要循環到 next[1] = 0 ,該字符的 next 值都爲 1 )

模式串T爲:                  “abcabac”
next數組(下標從1開始):011

第四個字符 ’a‘ :由於前一個字符 ‘c’ 的 next 值爲 1 ,取 T[1] = ‘a’ 和 ‘c’ 相比較,不相等,繼續;由於 next[1] = 0 ,結束。‘a’ 對應的 next 值爲 1 ;

模式串T爲:                  “abcabac”
next數組(下標從1開始):0111

第五個字符 ’b’ :由於前一個字符 ‘a’ 的 next 值爲 1 ,取 T[1] = ‘a’ 和 ‘a’ 相比較,相等,結束。 ‘b’ 對應的 next 值爲:1(前一個字符 ‘a’ 的 next 值) + 1 = 2 ;

模式串T爲:                  “abcabac”
next數組(下標從1開始):01112

第六個字符 ‘a’ :由於前一個字符 ‘b’ 的 next 值爲 2,取 T[2] = ‘b’ 和 ‘b’ 相比較,相等,所以結束。‘a’ 對應的 next 值爲:2 (前一個字符 ‘b’ 的 next 值) + 1 = 3 ;

模式串T爲:                  “abcabac”
next數組(下標從1開始):011123

第七個字符 ‘c’ :由於前一個字符 ‘a’ 的 next 值爲 3 ,取 T[3] = ‘c’ 和 ‘a’ 相比較,不相等,繼續;由於 next[3] = 1 ,所以取 T[1] = ‘a’ 和 ‘a’ 比較,相等,結束。‘a’ 對應的 next 值爲:1 ( next[3] 的值) + 1 = 2 ;

模式串T爲:                  “abcabac”
next數組(下標從1開始):0111232

算法實現:

void Next(char *str, int *next)
{
        int len = strlen(str);
        int i = 1;
        int j = 0;
        next[i] = 0;
        while(i < len)
        {
                if(j == 0 || str[i-1] == str[j-1])
                {
                        i++;
                        j++;
                        next[i] = j;
                }
                else
                {
                        j = next[j];
                }
        }
}


注意:在此程序中,next 數組使用的下標初始值爲 1 ,next[0] 沒有用到(也可以存放 next 數組的長度)。而串的存儲是從數組的下標 0 開始的,所以程序中爲 T[i-1] 和 T[j-1]。

基於next的KMP算法的實現

先看一下 KMP 算法運行流程(假設主串:ababcabcacbab,模式串:abcac)。

第一次匹配:

匹配失敗,i 指針不動,j = 1(字符‘c’的next值);

第二次匹配:

相等,繼續,直到:

匹配失敗,i 不動,j = 2 ( j 指向的字符 ‘c’ 的 next 值);

第三次匹配:

相等,i 和 j 後移,最終匹配成功。

使用普通算法,需要匹配 6 次;而使用 KMP 算法,則只匹配 3 次。

實現代碼:

int KMP(char *str1, char *str2)
{
        //都從1開始
        int i = 1;
        int j = 1;
        int next[10];
        Next(str2, next);
        while(i <= strlen(str1) && j <= strlen(str2))
        {
                if(str1[i-1] == str2[j-1])
                {
                        i++;
                        j++;
                }
                else
                {
                        j = next[j];
                }
        }
        if(j > strlen(str2))
        {
                return i - (int)strlen(str2);
        }
        return -1;
}


KMP算法完整代碼

#include <stdio.h>
#include <string.h>

void Next(char *str, int *next)
{
        int len = strlen(str);
        int i = 1;
        int j = 0;
        next[i] = 0;
        while(i < len)
        {
                if(j == 0 || str[i-1] == str[j-1])
                {
                        i++;
                        j++;
                        next[i] = j;
                }
                else
                {
                        j = next[j];
                }
        }
}

int KMP(char *str1, char *str2)
{
        //都從1開始
        int i = 1;
        int j = 1;
        int next[10];
        Next(str2, next);
        while(i <= strlen(str1) && j <= strlen(str2))
        {
                if(str1[i-1] == str2[j-1])
                {
                        i++;
                        j++;
                }
                else
                {
                        j = next[j];
                }
        }
        if(j > strlen(str2))
        {
                return i - (int)strlen(str2);
        }
        return -1;
}

int main()
{
        int pos = KMP("ababcabcacbab", "abcac");
        printf("the pos is %d\n", pos);
        return 0;
}

運行結果:

the pos is 6


升級版的next

注意:KMP 算法的關鍵在於 next 數組的確定,其實對於上邊的KMP算法中的next數組,不是最精簡的,還可以簡化。

例如:

模式串T:a b c a c
    next  :0 1 1 1 2

在模式串“abcac”中,有兩個字符 ‘a’,我們假設第一個爲 a1,第二個爲 a2。在程序匹配過程中,如果 j 指針指向 a2 時匹配失敗,那麼此時,主串中的 i 指針不動,j 指針指向 a1 ,很明顯,由於 a1==a2,而 a2!=S[i],所以 a1 也肯定不等於 S[i]。

爲了避免不必要的判斷,需要對 next 數組進行精簡,對於“abcac”這個模式串來說,由於 T[4] == T[next[4]] ,所以,可以將next數組改爲:

模式串T:a b c a c
    next  :0 1 1 0 2

這樣簡化,如果匹配過程中由於 a2 匹配失敗,那麼也不用再判斷 a1 是否匹配,因爲肯定不可能,所以直接繞過 a1,進行下一步。

實現代碼:

void Next(char *str, int *next)
{
        int len = strlen(str);
        int i = 1;
        int j = 0;
        next[i] = 0;
        while(i < len)
        {
                if(j == 0 || str[i-1] == str[j-1])
                {
                        i++;
                        j++;
                        if(str[i-1] != str[j - 1])
                        {
                                next[i] = j;
                        }
                        else
                        {
                                next[i] = next[j];
                        }
                }
                else
                {
                        j = next[j];
                }
        }
}


使用精簡過後的 next 數組在解決例如模式串爲“aaaaaaab”這類的問題上,會減少很多不必要的判斷次數,提高了KMP算法的效率。

例如:精簡前爲 next1,精簡後爲 next2:

模式串:a a a a a a a b
  next1:0 1 2 3 4 5 6 7
  next2:0 0 0 0 0 0 0 7

總結

KMP 算法,之所以比 BF 算法快的根本原因在於:KMP 算法其實也和 BF 算法一樣,都是從主串開頭開始匹配,但是在匹配過程中,KMP算法記錄了一些必要的信息。根據這些信息,在後續的匹配過程中,跳過了一些無意義的匹配過程。



發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章