LeetCode 718 最长重复子数组

给两个整数数组 A 和 B ，返回两个数组中公共的、长度最长的子数组的长度。

A: [1,2,3,2,1]
B: [3,2,1,4,7]

数组+长度，这种问题看来很多都可以采用二分搜索算法，因为长度有一个范围。

搜索可能的长度，针对某个长度的字符串，找到原字符串所有子串对应的哈希值。再去哈希另一个字符串。

from typing import *


class Solution:
    def findLength(self, A: List[int], B: List[int]) -> int:
        res = 0
        lo, hi = 0, min(len(A), len(B))
        while lo <= hi:
            mid = (lo + hi) // 2
            if self.check(A, B, mid):
                res = mid
                lo = mid + 1
            else:
                hi = mid - 1
        return res

    def check(self, A, B, len1):
        mod = 10 ** 9 + 9
        base = 113
        mult = int(base ** (len1 - 1))
        val_set = set()
        val1 = 0
        for j in range(len1):
            val1 = (val1 * base + A[j]) % mod
        val_set.add(val1)
        for i in range(1, len(A) - len1 + 1):
            val1 = ((val1 - A[i - 1] * mult % mod) * base + A[i + len1 - 1]) % mod
            val_set.add(val1)
        val1 = 0
        for j in range(len1):
            val1 = (val1 * base + B[j]) % mod
        if val1 in val_set:
            return True
        for i in range(1, len(B) - len1 + 1):
            val1 = ((val1 - B[i - 1] * mult % mod) * base + B[i + len1 - 1]) % mod
            if val1 in val_set:
                return True
        return False