从零开始复习数据结构 python leetcode 5.最长回文子串

• 时间复杂度: O(n)
• 空间复杂度: O(n)

class Solution:
    def expand(self, s, left, right):
        """
        扩展左右边界以找到以当前位置为中心的最大回文半径。
        """
        while left >= 0 and right < len(s) and s[left] == s[right]:
            left -= 1
            right += 1
        return (right - left - 2) // 2  # 回文半径

    def longestPalindrome(self, s: str) -> str:
        # 预处理字符串，使得处理偶数长度的回文子串变得统一
        s = '#' + '#'.join(list(s)) + '#'

        # 初始化变量
        arm_len = []  # 存储每个位置上的回文半径
        right = -1  # 最右侧回文的右边界
        j = -1  # 最右侧回文的中心索引
        end, start = -1, 0  # 最长回文子串的开始和结束索引

        # 主循环遍历每个位置
        for i in range(len(s)):
            if right >= i:
                # 如果当前位置在最右侧回文的范围内，则根据镜像位置计算最小回文半径
                i_sym = 2 * j - i
                min_arm_len = min(arm_len[i_sym], right - i)
                cur_arm_len = self.expand(s, i - min_arm_len, i + min_arm_len)
            else:
                # 如果当前位置不在最右侧回文的范围内，则直接从当前位置扩展
                cur_arm_len = self.expand(s, i, i)

            # 更新回文半径列表
            arm_len.append(cur_arm_len)

            # 如果当前位置的回文半径加上中心索引超过了最右侧回文的右边界，则更新相关信息
            if i + cur_arm_len > right:
                j = i
                right = i + cur_arm_len

            # 如果当前回文的长度大于已知最长回文的长度，则更新最长回文的信息
            if 2 * cur_arm_len + 1 > end - start:
                start = i - cur_arm_len
                end = i + cur_arm_len

        # 返回最长回文子串
        # 注意由于预处理添加了'#'字符，所以这里需要去掉这些字符
        return s[start+1:end+1:2]

solution = Solution()
s = "bb"
result = solution.longestPalindrome(s)
print(result)

学算法真的很难，有时候一两个小时也不能完全明白一道题