131. 分割回文串 - 力扣（LeetCode）

maybe_za2024-01-26 3:03

问题描述

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

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

输入示例

bash 复制代码

s = "aab"

输出示例

bash 复制代码

[["a","a","b"],["aa","b"]]

解题思路

我们使用回溯、深度优先遍历的思想，使用 ans 记录路径，使用 ret 记录路径组合结果，使用 f 数组记录是否回文，使用 n 记录字符串总数量。以下为核心递归逻辑，i 表示分割的开始位置：

如果 i==n，表示已经分割到结尾，则将路径结果 ans 放入 ret。
否则从 i 开始遍历分割，如果回文，将从 i 到 j 的子串放入路径
继续从下一个位置开始分割递归
执行回溯过程

解题代码

java 复制代码

class Solution {
    boolean[][] f;
    List<List<String>> ret = new ArrayList<>();
    List<String> ans = new ArrayList<>();
    int n;

    public List<List<String>> partition(String s) {
        n = s.length();
        f = new boolean[n][n];
        for(int i = 0; i < n; i++) {
            Arrays.fill(f[i], true);
        }
        for(int i = n-1; i >= 0; i--) {
            for(int j = i+1; j < n; j++) {
                f[i][j] = (s.charAt(i) == s.charAt(j)) && f[i+1][j-1];
            }
        }
        dfs(s, 0);
        return ret;
    }

    public void dfs(String s, int i) {
        if(i == n) {
            ret.add(new ArrayList<String>(ans));
            return;
        }
        for(int j = i; j < n; j++) {
            if(f[i][j]) {
                ans.add(s.substring(i, j + 1));
                dfs(s, j + 1);
                ans.remove(ans.size() - 1);
            }
        }
    }
}