LeetCode 131. 分割回文串 (回溯)

Description

給定一個字符串 s，將 s 分割成一些子串，使每個子串都是迴文串。

返回 s 所有可能的分割方案。

示例:

輸入: "aab"
輸出:
[
  ["aa","b"],
  ["a","a","b"]
]

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/palindrome-partitioning
著作權歸領釦網絡所有。商業轉載請聯繫官方授權，非商業轉載請註明出處。

Solution

參考題解畫出回溯樹容易理解代碼。

class Solution:
    def partition(self, s: str) -> List[List[str]]:
        res = []

        def backtrack(i, tmp):
            if i == len(s): res.append(tmp[:])
            for j in range(i, len(s)):
                if s[i:j+1] == s[i:j+1][::-1]:
                    tmp.append(s[i:j+1])
                    backtrack(j+1,tmp)
                    tmp.pop()
        backtrack(0, [])
        # print(res)
        return res