leetcode131 dp+回溯+vector代替stack

题目:
给定一个字符串 s,将 s 分割成一些子串,使每个子串都是回文串。
返回 s 所有可能的分割方案。

class Solution {
public:
    vector<vector<string>> partition(string s)
    {
        vector< vector<int> > dp = isPalindrome(s, 0, s.size());
        vector<vector<string>> result;
        vector<string> strs = {};
        vector<pair<int, int>> lr = {};
        int end = 0;
        for (int i = 0; i < s.size(); ++i) {
            if (dp[0][i] == 1) {
                lr.push_back({ 0, i });
            }
        }
        while (!lr.empty()) {
            int l = lr.back().first, r = lr.back().second;
            lr.pop_back();
            int len = r - l + 1;
            while (r < end) {
                end -= strs.back().size();
                strs.pop_back();
            }
            strs.push_back(s.substr(l, r - l + 1));
            end += len;
            if (r == s.size() - 1) {
                result.push_back(strs);
                end -= strs.back().size();
                strs.pop_back();
            }
            for (int i = r + 1; i < s.size(); ++i) {
                if (dp[r + 1][i] == 1) {
                    lr.push_back({ r + 1, i });
                }
            }
        }
        return result;
    };

private:
    vector<vector<int>> isPalindrome(string &s, int left, int right)
    {
        int len = s.size();
        vector< vector<int> > dp(len, vector<int>(len, 0));
        for (int i = 0; i < len; i++) {
            dp[i][i] = 1;
            if ((i + 1 < len) && s[i] == s[i + 1]) {
                dp[i][i + 1] = 1;
            }
        }
        for (int i = len - 1; i >= 0; i--) {
            for (int j = i + 1; j < len; j++) {
                if ((s[j] == s[i]) && (i + 1 < len) && (j - 1) >= 0 && (i + 2 <= j)) {
                    dp[i][j] = dp[i + 1][j - 1];
                }
            }
        }
        return dp;
    }
};
发布了33 篇原创文章 · 获赞 0 · 访问量 624

猜你喜欢

转载自blog.csdn.net/qq_28133013/article/details/103969769