【Lintcode】1159. Longest Common Prefix III

题目地址：

https://www.lintcode.com/problem/longest-common-prefix-iii/description

给定一个字符串数组$A$，再给定若干个询问，每次询问某两个字符串$A[i]$和$A[j]$的最长公共前缀的长度。

可以先预处理一下$A$里每个字符串的前缀哈希数组，然后求最长公共前缀的时候，就可以用二分来做了。对于$A[i]$和$A[j]$的前缀哈希数组$h[i]$和$h[j]$，实际上就是要找最大的$k$使得$h[i][k]=h[j][k]$，这样最长公共前缀的长度就是$k$。代码如下：

public class Solution {
    /**
     * @param arr:   string array
     * @param query: query array
     * @return: return  LCP ans array
     */
    public int[] queryLCP(String[] arr, int[][] query) {
        // write your code here
        // 预处理一下arr[i]的前缀哈希数组
        long[][] hash = new long[arr.length][];
        for (int i = 0; i < arr.length; i++) {
            String s = arr[i];
            hash[i] = new long[s.length() + 1];
            long p = 131;
            for (int j = 0; j < s.length(); j++) {
                hash[i][j + 1] = hash[i][j] * p + s.charAt(j);
            }
        }
        
        int[] res = new int[query.length];
        for (int i = 0; i < query.length; i++) {
            res[i] = lcp(hash, query[i][0], query[i][1]);
        }
        
        return res;
    }
    
    // 用二分求arr[i]和arr[j]的最长公共前缀长度
    private int lcp(long[][] hash, int i, int j) {
        int l = 0, r = Math.min(hash[i].length, hash[j].length) - 1;
        while (l < r) {
            int m = l + (r - l + 1 >> 1);
            if (hash[i][m] == hash[j][m]) {
                l = m;
            } else {
                r = m - 1;
            }
        }
        
        return l;
    }
}

时间复杂度$O(\sum l_{s_i}+n\log l)$，$n$是询问次数，$l$是最长字符串长度。