力扣131. 分割回文串

题目链接

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

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

---

分析：
最坏的情况是所有字母都相同，类似于aaaaaaaa，那么这8个a就有七个位置可以插入，因为全都一样，所以无论怎么分割都满足每个子串都是回文串，那结果就有2^n种。所以最坏情况是指数级别。那可以直接搜索。

搜索思路：
按找每一段来搜索，起点为u，终点从u枚举到最后，如果能形成回文串，那就是一种答案
判断回文串可以预处理，利用一个递推式

class Solution {
    List<List<String>> res = new ArrayList<>();
    List<String> path = new ArrayList<>();
    boolean[][] f;

    public List<List<String>> partition(String s) {
        int n = s.length();
        f = new boolean[n][n];
        // 递推式，通过f[i][j] = true表示i ~ j 这一段是回文串
        for (int j = 0; j < n; j ++ )
            for (int i = 0; i <= j; i ++ )
                if (i == j) f[i][j] = true; // 如果只有一个字符一定是
                else if (s.charAt(i) == s.charAt(j)) { // 否则通过递推式，如果两个字符串相同
                    if (i + 1 == j || f[i + 1][j - 1]) f[i][j] = true; // 成立的条件是要么只有他们俩组成回文串，要么他们俩中间的那部分字符串也已经是回文串了
                }
        dfs(s, 0);
        return res;
    }

    void dfs(String s, int u) {
        if (u == s.length()) {
            res.add(new ArrayList(path));
            return;
        }
        for (int i = u; i < s.length(); i ++ ) { // 新的一段从u开始，u之前已经是分成几部分的回文串了
            if (f[u][i]) {
                path.add(s.substring(u, i + 1)); // u ~ i也是回文串，加入答案
                dfs(s, i + 1); // 搜索下一段
                path.remove(path.size() - 1); // 回溯
            }
        }
    }

}