剑指 Offer II 086. 分割回文子字符串

题目链接

力扣

题目描述

给定一个字符串 s ，请将 s 分割成一些子串，使每个子串都是 回文串 ，返回 s 所有可能的分割方案。

回文串 是正着读和反着读都一样的字符串。

示例 1：

输入：s = "google"
输出：[["g","o","o","g","l","e"],["g","oo","g","l","e"],["goog","l","e"]]
示例 2：

输入：s = "aab"
输出：[["a","a","b"],["aa","b"]]
示例 3：

输入：s = "a"
输出：[["a"]]
 

提示：

1 <= s.length <= 16
s 仅由小写英文字母组成

解题思路

首先用中心扩展方式找到所有的回文字符串存储在字典中，键为左指针，值为右指针组成的列表

使用回溯法构造所有的回文子串：backtrack(l)表示对s[l:n]构造回文分割方案，此时：
（1）若l=n，收集结果；
（2）否则从dic[l]中选择元素r，将s[l:(r+1)]加入stack，并执行backtrack(r+1);

代码

Python

class Solution:
    def partition(self, s: str) -> list[list[str]]:
        n = len(s)
        dic = {i: [] for i in range(n)}
        for c in range(2 * n - 1):
            l = c // 2
            r = l + c % 2
            while l >= 0 and r < n and s[l] == s[r]:
                dic[l].append(r)
                l -= 1
                r += 1

        ans = []
        path = []

        def backtrace(l: int) -> None:
            if l == n:
                ans.append(path[:])
                return
            for r in dic[l]:
                path.append(s[l:r + 1])
                backtrace(r+1)
                path.pop()
        backtrace(0)
        return ans