【LeetCode-回溯】分割回文串

时间:2020-07-11
本文章向大家介绍【LeetCode-回溯】分割回文串,主要包括【LeetCode-回溯】分割回文串使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。

题目描述

给定一个字符串 s,将 s 分割成一些子串,使每个子串都是回文串。

返回 s 所有可能的分割方案。
示例:

输入: "aab"
输出:
[
  ["aa","b"],
  ["a","a","b"]
]

题目链接: https://leetcode-cn.com/problems/palindrome-partitioning/

思路

由于是求所有可能分割情况,所以使用回溯来做。代码如下:

class Solution {
public:
    vector<vector<string>> partition(string s) {
        if(s.empty()) return {};

        vector<vector<string>> ans;
        vector<string> cur;
        int start = 0;
        dfs(s, start, cur, ans);
        return ans;
    }

    void dfs(string s, int start, vector<string> cur, vector<vector<string>>& ans){
        if(start>=s.size()){
            ans.push_back(cur);
            return;
        }

        for(int i=start; i<s.size(); i++){
            string sub = s.substr(start, i-start+1);
            if(judge(sub)){
                cur.push_back(sub);
                dfs(s, i+1, cur, ans);
                cur.pop_back();
            }
        }
    }

    /*判断s是否是回文串,是返回true,不是返回false*/
    bool judge(string s){
        if(s.size()==1) return true;
        int left = 0;
        int right = s.size()-1;
        while(left<right){
            if(s[left]!=s[right]) return false;
            left++;
            right--;
        }
        return true;
    }
};

使用下面的写法(多加了几个引用&)可以将速度提升 4 倍:

class Solution {
public:
    vector<vector<string>> partition(string s) {
        if(s.empty()) return {};

        vector<string> cur;
        vector<vector<string>> ans;
        int start = 0;
        dfs(s, start, cur, ans);
        return ans;
    }

    void dfs(string& s, int start, vector<string>& cur, vector<vector<string>>& ans){
        if(start>=s.size()){
            ans.push_back(cur);
            return;
        }

        for(int i=start; i<s.size(); i++){
            string sub = s.substr(start, i-start+1);
            if(judge(sub)){
                cur.push_back(sub);
                dfs(s, i+1, cur, ans);
                cur.pop_back();
            }
        }
    }

    /*判断s是否是回文串,是返回true,不是返回false*/
    bool judge(string& s){
        if(s.size()==1) return true;
        int left = 0;
        int right = s.size()-1;
        while(left<right){
            if(s[left]!=s[right]) return false;
            left++;
            right--;
        }
        return true;
    }
};

原文地址:https://www.cnblogs.com/flix/p/13285615.html