131.分割回文串
给定一个字符串 s,将 s 分割成一些子串,使每个子串都是回文串。
返回 s 所有可能的分割方案。
示例: 输入: "aab" 输出: [ ["aa","b"], ["a","a","b"] ]
class Solution { List<List<String>> ans=new ArrayList<>(); List<String> path=new ArrayList<>(); public List<List<String>> partition(String s) { process(s,0); return ans; } private void process(String s,int index){ if(index==s.length()){ ans.add(new ArrayList<>(path)); } for(int i=index;i<s.length();i++){ String cur=s.substring(index,i+1); if(isHuiWen(cur)){ path.add(cur); process(s,i+1); path.remove(path.size()-1); } } } private boolean isHuiWen(String str){ int l=0; int r=str.length()-1; while(l<=r){ if(str.charAt(l)!=str.charAt(r)){ return false; } l++; r--; } return true; } }
思路
本题这涉及到两个关键问题:
- 切割问题,有不同的切割方式
- 判断回文
回溯究竟是如何切割字符串呢?
分析一下切割,其实切割问题类似组合问题。
例如对于字符串abcdef:
- 组合问题:选取一个a之后,在bcdef中再去选取第二个,选取b之后在cdef中在选组第三个.....。
- 切割问题:切割一个a之后,在bcdef中再去切割第二段,切割b之后在cdef中在切割第三段.....。
感受出来了不?
所以切割问题,也可以抽象为一颗树形结构,如图:
递归用来纵向遍历,for循环用来横向遍历,切割线(就是图中的红线)切割到字符串的结尾位置,说明找到了一个切割方法。
此时可以发现,切割问题的回溯搜索的过程和组合问题的回溯搜索的过程是差不多的。
回溯三部曲
- 递归函数参数
全局变量数组path存放切割后回文的子串,二维数组result存放结果集。 (这两个参数可以放到函数参数里)
本题递归函数参数还需要startIndex,因为切割过的地方,不能重复切割,和组合问题也是保持一致的。
void process (String s, int startIndex,List<List<String>> res,List<String> path)
- 递归函数终止条件
从树形结构的图中可以看出:切割线切到了字符串最后面,说明找到了一种切割方法,此时就是本层递归的终止终止条件。
那么在代码里什么是切割线呢?
在处理组合问题的时候,递归参数需要传入startIndex,表示下一轮递归遍历的起始位置,这个startIndex就是切割线。
所以终止条件代码如下:
void process (string s, int startIndex) {
// 如果起始位置已经大于s的大小,说明已经找到了一组分割方案了
if (startIndex >= s.size()) {
result.push(path);
return;
}
}
- 单层搜索的逻辑
来看看在递归循环,中如何截取子串呢?
在for (int i = startIndex; i < s.size(); i++)
循环中,我们 定义了起始位置startIndex,那么 [startIndex, i] 就是要截取的子串。
首先判断这个子串是不是回文,如果是回文,就加入在List<string> path
中,path用来记录切割过的回文子串。
代码如下:
for (int i = startIndex; i < s.size(); i++) {
if (isPalindrome(s, startIndex, i)) { // 是回文子串
// 获取[startIndex,i]在s中的子串
string str = s.substr(startIndex, i - startIndex + 1);
path.push(str);
} else {
// 如果不是则直接跳过
continue;
}
process(s, i + 1); // 寻找i+1为起始位置的子串
path.remove(paht.size-1); // 回溯过程,弹出本次已经填在的子串
}
注意切割过的位置,不能重复切割,所以,process(s, i + 1); 传入下一层的起始位置为i + 1。
判断回文子串
最后我们看一下回文子串要如何判断了,判断一个字符串是否是回文。
可以使用双指针法,一个指针从前向后,一个指针从后先前,如果前后指针所指向的元素是相等的,就是回文字符串了。
bool isPalindrome(String s, int start, int end) {
for (int i = start, j = end; i < j; i++, j--) {
if (s[i] != s[j]) {
return false;
}
}
return true;
}
总结
那么难究竟难在什么地方呢?
我列出如下几个难点:
- 切割问题可以抽象为组合问题
- 如何模拟那些切割线
- 切割问题中递归如何终止
- 在递归循环中如何截取子串
- 如何判断回文
class Solution { List<List<String>> res=new ArrayList<>(); List<String> path=new ArrayList<>(); public List<List<String>> partition(String s) { process(s,0); return res; } private void process(String s,int startIndex){ if(startIndex==s.length()){ res.add(new ArrayList<>(path)); } for(int i=startIndex;i<s.length();i++){ String curStr=s.substring(startIndex,i+1); if(isHuiWen(curStr,0,curStr.length()-1)){ path.add(curStr); process(s,i+1); path.remove(path.size()-1); } } } private Boolean isHuiWen(String s,int start,int end){ if(s==null||s.length()<=1){ return true; } while(start<=end){ if(s.charAt(start)!=s.charAt(end)){ return false; } start++; end--; } return true; } }