730. 统计不同回文子字符串

题目描述

730. 统计不同回文子字符串

给定一个字符串 S，找出 S 中不同的非空回文子序列个数，并返回该数字与 10^9 + 7 的模。

通过从 S 中删除 0 个或多个字符来获得子字符序列。

如果一个字符序列与它反转后的字符序列一致，那么它是回文字符序列。

如果对于某个 i，A_i != B_i，那么 A_1, A_2, … 和 B_1, B_2, … 这两个字符序列是不同的。

示例 1：

输入：
S = 'bccb'
输出：6
解释：
6 个不同的非空回文子字符序列分别为：'b', 'c', 'bb', 'cc', 'bcb', 'bccb'。
注意：'bcb' 虽然出现两次但仅计数一次。

示例 2：

输入：
S = 'abcdabcdabcdabcdabcdabcdabcdabcddcbadcbadcbadcbadcbadcbadcbadcba'
输出：104860361
解释：
共有 3104860382 个不同的非空回文子字符序列，对 10^9 + 7 取模为 104860361。

提示：

字符串 S 的长度将在[1, 1000]范围内。
每个字符 S[i] 将会是集合 {‘a’, ‘b’, ‘c’, ‘d’} 中的某一个。

解题思路

dp[i][j]定义为在区间i，j内含有多少个回文子串。
如果 S[i] == S[j]，这时我们需要判断[i, j]这一段中有多少字符与S[i]不相等
	如果中间没有和S[i]相同的字母，例如"aba"这种情况，dp[i][j] = dp[i + 1][j - 1] * 2 + 2;（dp[i][j] = dp[i + 1][j - 1] * 2 代表的dp[i + 1[j - 1]这一段可以独立存在，也可在外层包裹S[i],S[j]，所有需要x2，而2是代表“aa”和“a”）
	如果中间只有一个和S[i]相同的字母，就是"aaa"这种情况，dp[i][j] = dp[i + 1][j - 1] * 2 + 1;（x2与上面情况相同，加一单独计算"aa",而“a”在dp[i + 1][j - 1] 中计算过了）
	否则中间至少有两个和S[i]相同的字母，就是"aabaa"这种情况，dp[i][j] = dp[i + 1][j - 1] * 2 - dp[left + 1][right - 1];（left、right请见代码注释，dp[left + 1][right - 1]这一段重复计算了）
否则dp[i][j] = dp[i][j - 1] + dp[i + 1][j] - dp[i + 1][j - 1];

class Solution:
    def countPalindromicSubsequences(self, S: str) -> int:
        lens = len(S)
        dp = [[1 for _ in range(lens)] for _ in range(lens)]
        for i in range(lens-2,-1,-1):
            for j in range(i+1,lens):
                if S[i] == S[j]:
                    if j == i+1:
                        dp[i][j] = 2
                    else:
                        k1,k2 = -1,-1
                        for m in range(i+1,j):
                            if S[m] == S[i]:
                                k1 = m
                                break
                        for n in range(j-1,i,-1):
                            if S[n] == S[i]:
                                k2 = n
                                break
                        if k1 == k2 and k2 == -1:
                            dp[i][j] = 2 * dp[i+1][j-1] + 2
                        elif k1 == k2 and k1 != -1:
                            dp[i][j] = 2 * dp[i+1][j-1] + 1
                        else:   # k1,k2不相等
                            if k1 + 1 == k2:
                                dp[i][j] = 2 * dp[i+1][j-1]
                            else:
                                dp[i][j] = 2 * dp[i+1][j-1] - dp[k1+1][k2-1]         
                else:
                    if j == i+1:
                        dp[i][j] = 2
                    else:
                        dp[i][j] = dp[i][j-1] + dp[i+1][j] - dp[i+1][j-1]
                dp[i][j] %= 10**9 + 7
        return dp[0][lens-1]