Java: KMP匹配算法

Question

​

​

我试图在Java中实现上述算法。但是，我得到了一个超出范围的异常，我不知道如何解决这个问题。

我只是在逐行翻译psuedocode。

以下是代码：

public static int[] computePrefixFunction(String input) 
    {
        int[] pi = new int[input.length()];
        int k = 0;
        for (int q = 1; q < input.length(); q++) {
            char target = input.charAt(q);
        while (k > 0 && input.charAt(k) != target) k = pi[k - 1];
        if (input.charAt(k) == target) k++;
        pi[q] = k;
    }
    return pi;
}

public static Queue<Integer> KMPMatcher(String T, String P) 
{
    int n = T.length();
    int m = P.length();
    int[] pi = computePrefixFunction(P);
    int q = 0;
    Queue<Integer> Q = new LinkedList<>();
    for(int i = 0; i < n; i++) 
    {
        while(q > 0 && P.charAt(q+1) != T.charAt(i)) 
            q = pi[q];
        if(P.charAt(q+1) == T.charAt(i)) 
            q = q + 1;
        if(q == m-1) // you match it when q reaches size of pattern -1. :)
        {
            Q.add(i-m+1); // Change it as well.
            q = pi[q];
        }     
    }
    return Q;
}

public static void main(String[] args) {
    System.out.println(KMPMatcher("bdacabdacb","bda"));
}

编辑:我已经用piyush实现更新了代码，下面纠正了我的一些问题。然而，还有另一个问题。

我使用以下方法测试了KMPMatcher：

1) System.out.println(KMPMatcher("bacabab","bab")); // returned2,4

2) System.out.println(KMPMatcher("bdacabdacb","bab")); // returned3

数字1应该只返回4，第2号应该只返回一个空列表。为什么会发生这种情况？我试图用这些输入绘制跟踪，并将其与psuedocode进行比较。我认为这与if(q==m-1)中的索引有关(因为它没有将正确的东西与psuedocode版本进行比较？)我也不知道该怎么解决。有什么帮助吗？

Answer 1

问题在if语句中。不应该是if (q == m-1)。

public static int[] computePrefixFunction(String input) 
    {
        int[] pi = new int[input.length()];
        int k = 0;
        for (int q = 1; q < input.length(); q++) {
            char target = input.charAt(q);
        while (k > 0 && input.charAt(k) != target) k = pi[k - 1];
        if (input.charAt(k) == target) k++;
        pi[q] = k;
    }
    return pi;
}

   public static Queue<Integer> KMPMatcher(String T, String P) 
{
    int n = T.length();
    int m = P.length();
    int[] pi = computePrefixFunction(P);
    int q = 0;
    Queue<Integer> Q = new LinkedList<>();
    for(int i = 0; i < n; i++) 
    {
        while(q > 0 && P.charAt(q) != T.charAt(i)) 
            q = pi[q-1];
        if(P.charAt(q) == T.charAt(i)) 
            q++;
        {
            Q.add(i-q+1); // Change it.
            q = pi[q-1];
        }     
    }
    return Q;
}

public static void main(String[] args) {
    System.out.println(KMPMatcher("bdacabdacb","bda"));
}