如何在Rabin Karp算法的滚动哈希中加入mod?

时间:2018-06-12 14:46:10

标签: string algorithm string-matching rabin-karp

我正在尝试用mod实现Rabin Karp算法。我正在使用的哈希函数是:

H1= c1*a^k-1 + c2*a^k-2 +c3*a^k-3 +…+ck*a^0

这里cx是字符的ASCII值。滚动它我首先通过减去它来删除第一个项,然后乘以a并通过将它乘以^ 0来添加新项。

现在的问题是处理我使用mod操作的大值,但这样做我无法正确滚动它。我的代码如下:

public class RabinKarp {
private static final int base = 26;
private static final int mod = 1180637;

public static void main(String[] args) {
    String text = "ATCAAGTTACCAATA";
    String pattern = "ATA";
    char[] textArr = text.toCharArray();
    char[] patternArr = pattern.toCharArray();
    System.out.println(getMatchingIndex(textArr, patternArr));
}

public static int getMatchingIndex(char[] textArr, char[] patternArr) {
    int n = textArr.length;
    int m = patternArr.length;
    int patternHash = getHashForPatternSize(patternArr, m);
    int textHash = getHashForPatternSize(textArr, m);
    for(int i = 0; i < n-m; i++) {
        if(patternHash == textHash && checkMatch(textArr, patternArr, i, m))
            return i;
        textHash = rollingHash(textArr, textHash, i, m);    
    }
    return -1;
}

public static boolean checkMatch(char[] textArr, char[] patternArr, int i, int m) {
    for(int j = 0; j < m; j++,i++) {
        if(textArr[i] != patternArr[j])
            return false;
    }
    return true;
}

public static int rollingHash(char[] textArr, int textHash, int i, int m) {
    return (textHash * base - modularExponentiation(base, m, mod) * (int)textArr[i] + (int) textArr[i+m])%mod;
}

public static int getHashForPatternSize(char[] arr, int m) {
    int hash = 0;
    for(int i = 0, p = m; i < m; i++, p--) {
        hash = (hash%mod + calcHash(arr[i], p)%mod)%mod;
    }
    return hash;
}

public static int calcHash(char alphabet, int p) {
    return (((int) alphabet)%mod * modularExponentiation(base, p, mod)%mod)%mod;
}

public static int modularExponentiation(int base, int p, int mod) {
    if(p == 0)
        return 1;
    if(p%2 == 0)
        return modularExponentiation((base*base)%mod, p/2, mod);
    else
        return (base*modularExponentiation((base*base)%mod, (p-1)/2, mod))%mod;
}
}

问题是textHashpatternHash在任何时候都不匹配。我确信问题出在mod操作上。任何人都可以告诉如何使用mod以及正确使用滚动哈希。我会非常感激。

1 个答案:

答案 0 :(得分:0)

计算Rabin-Karp滚动哈希的常用方法是考虑big-endian顺序的字符,而不是你的little-endian解决方案。这使得算法更容易,因为它避免了除法。模块划分非常重要,您不能简单地将其实现为(p/q)%b

如果我们将滚动哈希视为

H0…k-1 = (c0*ak-1 + c1*ak-2 + c2*ak-3 …+… ck-1*a0) mod b

然后下一个词是:

H1…k   = (         c1*ak-1 + c2*ak-2 …+… ck-1*a1 + ck*a0) mod b

我们可以很容易地看到

H1…k   = (a * H0…k-1 - c0*ak + ck) mod b

如果我们预先计算m == ak mod b,则会变为:

H1…k   = (a * H0…k-1 - m * c0 + ck) mod b

每次迭代的工作量要少得多,而且根本不依赖于除法。