题目
类型:二分查找、字符串
解题思路
题目要求得「能取得最大长度的任一方案」,首先以「最大长度」为分割点的数轴具有「二段性」:
- 小于等于最大长度方案均存在(考虑在最大长度方案上做删减)
- 大于最大长度的方案不存在。
二分范围为 [0, n],关键在于如何 check
函数,即实现「检查某个长度 len 作为最大长度,是否存在合法方案」。
对于常规做法而言,可枚举每个位置作为起点,得到长度为 len 的子串,同时使用 Set
但是该做法实现的 check
并非线性,子串的生成和存入容器的时执行的哈希函数执行均和子串长度相关,复杂度是 O(n * len)。
可以通过「字符串哈希」进行优化,
在二分前先通过 O(n) 的复杂度预处理出哈希数组,从而确保能够在 check 时能够 O(1) 得到某个子串的哈希值,最终将 check 的复杂度降为 O(n)。
代码
class Solution {
long[] h, p;
public String longestDupSubstring(String s) {
int P = 1313131, n = s.length();
h = new long[n + 10]; p = new long[n + 10];
p[0] = 1;
for (int i = 0; i < n; i++) {
p[i + 1] = p[i] * P;
h[i + 1] = h[i] * P + s.charAt(i);
}
String ans = "";
int l = 0, r = n;
while (l < r) {
int mid = l + r + 1 >> 1;
String t = check(s, mid);
if (t.length() != 0) l = mid;
else r = mid - 1;
ans = t.length() > ans.length() ? t : ans;
}
return ans;
}
String check(String s, int len) {
int n = s.length();
Set<Long> set = new HashSet<>();
for (int i = 1; i + len - 1 <= n; i++) {
int j = i + len - 1;
long cur = h[j] - h[i - 1] * p[j - i + 1];
if (set.contains(cur)) return s.substring(i - 1, j);
set.add(cur);
}
return "";
}
}