给定一个字符串 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); // 回溯
}
}
}
}