【算法27天:Day27】第七章回溯算法 LeetCode 分割回文串(131)

87 阅读2分钟

题目三:

image.png

解法一:(回溯)

解题思路:其实切割问题类似组合问题

例如对于字符串abcdef:

  • 组合问题:选取一个a之后,在bcdef中再去选取第二个,选取b之后在cdef中在选组第三个.....。
  • 切割问题:切割一个a之后,在bcdef中再去切割第二段,切割b之后在cdef中在切割第三段.....。

所以切割问题,也可以抽象为一棵树形结构,如图:

131.分割回文串

递归用来纵向遍历,for循环用来横向遍历,切割线(就是图中的红线)切割到字符串的结尾位置,说明找到了一个切割方法。

此时可以发现,切割问题的回溯搜索的过程和组合问题的回溯搜索的过程是差不多的。

回溯三部曲

  • 递归函数参数

全局变量数组path存放切割后回文的子串,二维数组result存放结果集。 (这两个参数可以放到函数参数里)

本题递归函数参数还需要startIndex,因为切割过的地方,不能重复切割,和组合问题也是保持一致的。

回溯算法:求组合总和(二) (opens new window)中我们深入探讨了组合问题什么时候需要startIndex,什么时候不需要startIndex。

代码如下:

let result = []
let path = []
const backtracking = function(s, startIndex)
  • 递归函数终止条件

131.分割回文串

从树形结构的图中可以看出:切割线切到了字符串最后面,说明找到了一种切割方法,此时就是本层递归的终止终止条件。

那么在代码里什么是切割线呢?

在处理组合问题的时候,递归参数需要传入startIndex,表示下一轮递归遍历的起始位置,这个startIndex就是切割线。

所以终止条件代码如下:

const backtracking = function(s, startIndex) {
// 如果起始位置已经大于s的大小,说明已经找到了一组分割方案了
    if (startIndex >= s.length) {
        result.push([...path])
        return
    }
}
  • 单层搜索的逻辑

来看看在递归循环,中如何截取子串呢?

for (int i = startIndex; i < s.length; i++)循环中,我们 定义了起始位置startIndex,那么 [startIndex, i] 就是要截取的子串。

首先判断这个子串是不是回文,如果是回文,就加入在path中,path用来记录切割过的回文子串。

代码如下:

for (int i = startIndex; i < s.length; i++) {
    if (isPalindrome(s, startIndex, i)) { // 是回文子串
        // 获取[startIndex,i]在s中的子串
        string str = s.substr(startIndex, i - startIndex + 1);
        path.push(str);
    } else {                // 如果不是则直接跳过
        continue;
    }
    backtracking(s, i + 1); // 寻找i+1为起始位置的子串
    path.();        // 回溯过程,弹出本次已经填在的子串
}

注意切割过的位置,不能重复切割,所以,backtracking(s, i + 1); 传入下一层的起始位置为i + 1

#判断回文子串

最后我们看一下回文子串要如何判断了,判断一个字符串是否是回文。

可以使用双指针法,一个指针从前向后,一个指针从后先前,如果前后指针所指向的元素是相等的,就是回文字符串了。

那么判断回文代码如下:

var isPalindrome = function(s, start, end) {
    for (let i = start, j = end; i < j; i++, j--) {
        if(s[i] !== s[j]) {
            return false
        }
    }
    return true
}

完整代码:

var partition = function(s) {
    let result = []
    let path = []

    const backtracking = function(s, startIndex) {
// 如果起始位置已经大于s的大小,说明已经找到了一组分割方案了
        if (startIndex >= s.length) {
            result.push([...path])
            return
        }
        for (let i = startIndex; i < s.length; i++) {
            if (isPalindrome(s, startIndex, i)) { // 是回文子串
                // 获取[startIndex,i]在s中的子串
                let str = s.substr(startIndex, i - startIndex + 1)
                path.push(str)
            } else { // 不是回文,跳过
                continue
            }
            backtracking(s, i + 1) // 寻找i+1为起始位置的子串
            path.pop() // 回溯过程,弹出本次已经填入的子串
        }
    }
    backtracking(s, 0)
    return result
}; 
var isPalindrome = function(s, start, end) {
    for (let i = start, j = end; i < j; i++, j--) {
        if(s[i] !== s[j]) {
            return false
        }
    }
    return true
}

代码随想录解法:

/**
 * @param {string} s
 * @return {string[][]}
 */
const isPalindrome = (s, l, r) => {
    for (let i = l, j = r; i < j; i++, j--) {
        if(s[i] !== s[j]) return false;
    }
    return true;
}

var partition = function(s) {
    const res = [], path = [], len = s.length;
    backtracking(0);
    return res;
    function backtracking(i) {
        if(i >= len) {
            res.push(Array.from(path));
            return;
        }
        for(let j = i; j < len; j++) {
            if(!isPalindrome(s, i, j)) continue;
            path.push(s.slice(i, j + 1));
            backtracking(j + 1);
            path.pop();
        }
    }
};