力扣647. 回文子串- python代码

题目描述：

给定一个字符串，你的任务是计算这个字符串中有多少个回文子串。

具有不同开始位置或结束位置的子串，即使是由相同的字符组成，也会被视为不同的子串。

示例 1：

输入：“abc”
输出：3
解释：三个回文子串: “a”, “b”, “c”

示例 2：

输入：“aaa”
输出：6
解释：六个回文子串: “a”, “a”, “a”, “aa”, “aa”, “aaa”

解题思路：

方法一：暴力枚举。枚举出所有的子串，然后判断子串是否为回文字符串，时间复杂度为 $O(n^3)$ ，不推荐。

方法二：中心扩散。依次以每个字符和每两个相邻字符为中心，从中心向两边扩散，判断扩散出来的子串是否回文。时间复杂度为 $O(n^2)$。

方法三：动态规划。定义状态 $dp(i,j)$ 表示以 $i$ 开头，以 $j$ 结尾的子串是否为回文，状态转移方程为 $dp(i,j)=s(i)==s(j)\mathrm{and}dp(i+1,j-1)$，时间复杂度为 $O(n^2)$。

代码实现及复杂度分析：

Python 代码

方法二：

class Solution:
    def countSubstrings(self, s: str) -> int:
        n, res = len(s), 0
        for i in range(n):
            res += self.count(s, i, i)
            res += self.count(s, i, i + 1)
        return res
    
    def count(self, s: str, l: int, r: int) -> int:
        res = 0
        while l >= 0 and r < len(s) and s[l] == s[r]:
            l -= 1
            r += 1
            res += 1
        return res

时间复杂度为 $O(n^2)$，其中 $n$ 为字符串长度。

空间复杂度为 $O(1)$。

方法三：

class Solution:
    def countSubstrings(self, s: str) -> int:
        n, res = len(s), 0
        dp = [[False] * n for _ in range(n)]
        for j in range(n):
            for i in range(j + 1):
                if s[i] == s[j] and (j - i <= 2 or dp[i + 1][j - 1]):
                    dp[i][j] = True
                    res += 1
        return res

时间复杂度为 $O(n^2)$，其中 $n$ 为字符串长度。
空间复杂度为 $O(n^2)$，需要矩阵存储所有子串是否是回文子串的状态。