如何在Python中使用后缀数组进行字符串匹配

def build_suffix_array(text):
    suffixes = [(text[i:], i) for i in range(len(text))]  # 构建后缀数组
    suffixes.sort()  # 按照字典序排序
    return [suffix[1] for suffix in suffixes]  # 返回后缀数组

def search(text, pattern, suffix_array):
    # 利用二分查找在后缀数组中查找pattern
    left, right = 0, len(text)
    while left < right:
        mid = (left + right) // 2
        suffix = text[suffix_array[mid]:]
        if suffix.startswith(pattern):
            return True
        elif suffix < pattern:
            left = mid + 1
        else:
            right = mid
    return False

text = 'pidancode.com'
suffix_array = build_suffix_array(text)
pattern = 'code'
if search(text, pattern, suffix_array):
    print(f'{pattern} found in {text}')
else:
    print(f'{pattern} not found in {text}')