题目地址:
https://www.lintcode.com/problem/split-string/description
给定一个字符,要求将其分割为若干不相交的部分使得每个部分的子串长度小于等于 2 2 2。返回所有分割的情况。
思路是DFS。每次从字符串中某个位置开始枚举分割出一个或两个字符的子串,加入一个列表,然后从后一个位置开始进入下一层递归,直到分割到末尾时,将列表加入最终结果。代码如下:
import java.util.ArrayList;
import java.util.List;
public class Solution {
/*
* @param : a string to be split
* @return: all possible split string array
*/
public List<List<String>> splitString(String s) {
// write your code here
List<List<String>> res = new ArrayList<>();
dfs(s, 0, new ArrayList<>(), res);
return res;
}
// 从s的s[pos]开始,向后分割出一个或两个字符的子串,加入cur;res储存最后的答案
private void dfs(String s, int pos, List<String> cur, List<List<String>> res) {
// 如果pos到达了字符串尾,说明已经切割完毕,则将切割的结果加入res
if (pos == s.length()) {
res.add(new ArrayList<>(cur));
return;
}
// 开始枚举从pos后分割长度为1或2的子串,加入cur,并进入下一层递归
// 递归结束后要记得回溯,恢复现场
for (int i = pos; i < pos + 2 && i < s.length(); i++) {
cur.add(s.substring(pos, i + 1));
dfs(s, i + 1, cur, res);
cur.remove(cur.size() - 1);
}
}
}
时间复杂度 O ( n 2 n ) O(n2^n) O(n2n)(递归树是二叉树,并且深度为 n n n,同时,把cur加入res的时候要把cur拷贝一份),空间 O ( n ) O(n) O(n)。