LeetCode 力扣 131. 分割回文串

题目描述(中等难度)

给一个字符串,然后在任意位置切割若干次,保证切割后的每个字符串都是回文串。输出所有满足要求的切割结果。

解法一 分治

将大问题分解为小问题,利用小问题的结果,解决当前大问题。

这道题的话,举个例子。

aabb
先考虑在第 1 个位置切割,a | abb
这样我们只需要知道 abb 的所有结果,然后在所有结果的头部把 a 加入
abb 的所有结果就是 [a b b] [a bb]
每个结果头部加入 a,就是 [a a b b] [a a bb]

aabb
再考虑在第 2 个位置切割,aa | bb
这样我们只需要知道 bb 的所有结果,然后在所有结果的头部把 aa 加入
bb 的所有结果就是 [b b] [bb]
每个结果头部加入 aa,就是 [aa b b] [aa bb]

aabb
再考虑在第 3 个位置切割,aab|b
因为 aab 不是回文串,所有直接跳过

aabb
再考虑在第 4 个位置切割,aabb |
因为 aabb 不是回文串,所有直接跳过

最后所有的结果就是所有的加起来
[a a b b] [a a bb] [aa b b] [aa bb]

然后中间的过程求 abb 的所有结果,求 aab 的所有结果等等,就可以递归的去求。递归出口的话,就是空串的所有子串就是一个空的list 即可。

public List> partition(String s) {
    return partitionHelper(s, 0);
}

private List> partitionHelper(String s, int start) {
    //递归出口,空字符串
    if (start == s.length()) {
        List list = new ArrayList<>();
        List> ans = new ArrayList<>();
        ans.add(list);
        return ans;
    }
    List> ans = new ArrayList<>();
    for (int i = start; i < s.length(); i++) {
        //当前切割后是回文串才考虑
        if (isPalindrome(s.substring(start, i + 1))) {
            String left = s.substring(start, i + 1);
            //遍历后边字符串的所有结果,将当前的字符串加到头部
            for (List l : partitionHelper(s, i + 1)) {
                l.add(0, left);
                ans.add(l);
            }
        }

    }
    return ans;
}

private boolean isPalindrome(String s) {
    int i = 0;
    int j = s.length() - 1;
    while (i < j) {
        if (s.charAt(i) != s.charAt(j)) {
            return false;
        }
        i++;
        j--;
    }
    return true;
}

分治的话,一般情况下都可以利用动态规划的思想改为迭代的形式。递归就是压栈压栈,然后到达出口就出栈出栈出栈。动态规划就可以把压栈的过程省去,直接从递归出口往回考虑。之前做过很多题了,可以参考 77题、91 题、115 题 等等,都是一样的思想。这道题修改的话,看完解法二的优化后可以参考 这里 的代码。

解法二 分治优化

每次判断一个字符串是否是回文串的时候,我们都会调用 isPalindrome 判断。这就会造成一个问题,比如字符串 abbbba,期间我们肯定会判断 bbbb 是不是回文串,也会判断 abbbba 是不是回文串。判断 abbbba 是不是回文串的时候,在 isPalindrome 中依旧会判断中间的 bbbb 部分。而其实如果我们已经知道了 bbbb 是回文串,只需要判断 abbbba 的开头和末尾字符是否相等即可。

所以我们为了避免一些重复判断,可以用动态规划的方法,把所有字符是否是回文串提前存起来。

对于字符串 s

dp[i][j] 表示 s[i,j] 是否是回文串。

然后有 dp[i][j] = s[i] == s[j] && dp[i+1][j-1]

我们只需要两层 for 循环,从每个下标开始,考虑所有长度的子串即可。

boolean[][] dp = new boolean[s.length()][s.length()];
int length = s.length();
//考虑所有长度的子串
for (int len = 1; len <= length; len++) {
    //从每个下标开始
    for (int i = 0; i <= s.length() - len; i++) {
        int j = i + len - 1;
        dp[i][j] = s.charAt(i) == s.charAt(j) && (len < 3 || dp[i + 1][j - 1]);
    }
}

因为要保证 dp[i + 1][j - 1]i + 1 <= j - 1

i + 1 <= j - 1
把 j = i + len - 1 代入上式
i + 1 <= i + len - 1 - 1
化简得
len >= 3

所以为了保证正确,多加了 len < 3 的条件。也就意味着长度是 12 的时候,我们只需要判断 s[i] == s[j]

然后把 dp 传入到递归函数中即可。

public List> partition(String s) {
    boolean[][] dp = new boolean[s.length()][s.length()];
    int length = s.length();
    for (int len = 1; len <= length; len++) {
        for (int i = 0; i <= s.length() - len; i++) {
            int j = i + len - 1;
            dp[i][j] = s.charAt(i) == s.charAt(j) && (len < 3 || dp[i + 1][j - 1]);
        }
    }
    return partitionHelper(s, 0, dp);
}

private List> partitionHelper(String s, int start, boolean[][] dp) {
    if (start == s.length()) {
        List list = new ArrayList<>();
        List> ans = new ArrayList<>();
        ans.add(list);
        return ans;
    }
    List> ans = new ArrayList<>();
    for (int i = start; i < s.length(); i++) {
        if (dp[start][i]) {
            String left = s.substring(start, i + 1);
            for (List l : partitionHelper(s, i + 1, dp)) {
                l.add(0, left);
                ans.add(l);
            }
        }

    }
    return ans;
}

解法三 回溯

115 题 中考虑了分治、回溯、动态规划,这道题同样可以用回溯法。

回溯法其实就是一个 dfs 的过程,同样举个例子。

aabb
先考虑在第 1 个位置切割,a | abb
把 a 加入到结果中 [a]

然后考虑 abb
先考虑在第 1 个位置切割,a | bb
把 a  加入到结果中 [a a]

然后考虑 bb
先考虑在第 1 个位置切割,b | b
把 b 加入到结果中 [a a b] 

然后考虑 b
先考虑在第 1 个位置切割,b | 
把 b 加入到结果中 [a a b b] 

然后考虑空串
把结果加到最终结果中 [[a a b b]]

回溯到上一层 
考虑 bb
考虑在第 2 个位置切割,bb |
把 bb 加入到结果中 [a a bb] 

然后考虑 空串
把结果加到最终结果中 [[a a b b] [a a bb]]

然后继续回溯

可以看做下边的图做 dfs ,而每一层其实就是当前字符串所有可能的回文子串。

就是很经典的回溯法,一个 for 循环,添加元素,递归,删除元素。这里判断是否是回文串,我们就直接用 dp 数组。

public List> partition(String s) {
    boolean[][] dp = new boolean[s.length()][s.length()];
    int length = s.length();
    for (int len = 1; len <= length; len++) {
        for (int i = 0; i <= s.length() - len; i++) {
            dp[i][i + len - 1] = s.charAt(i) == s.charAt(i + len - 1) && (len < 3 || dp[i + 1][i + len - 2]);
        }
    }
    List> ans = new ArrayList<>();
    partitionHelper(s, 0, dp, new ArrayList<>(), ans);
    return ans;
}

private void partitionHelper(String s, int start, boolean[][] dp, List temp, List> res) {
    //到了空串就加到最终的结果中
    if (start == s.length()) {
        res.add(new ArrayList<>(temp));
    }
    //在不同位置切割
    for (int i = start; i < s.length(); i++) {
        //如果是回文串就加到结果中
        if (dp[start][i]) {
            String left = s.substring(start, i + 1);
            temp.add(left);
            partitionHelper(s, i + 1, dp, temp, res);
            temp.remove(temp.size() - 1);
        }

    }
}

这道题没有什么新内容了,就是分治、回溯、动态规划,很常规的题目了。

更多详细通俗题解详见 leetcode.wang 。

你可能感兴趣的:(LeetCode 力扣 131. 分割回文串)