1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

class Solution
{
public:
    std::vector<std::vector<std::string>> res; // 存储最终结果
    std::vector<std::string> path; // 当前路径
    std::vector<std::vector<bool>> f; // 动态规划存储回文判断结果

    vector<vector<string>> partition(string s)
    {
      int n = s.size();
      // 初始化回文判断数组
      f = std::vector<std::vector<bool>>(n, std::vector<bool>(n));

      // 动态规划预处理回文子串
      for (int j = 0; j < n; ++j)
        for (int i = 0; i <= j; ++i)
        {
          if (i == j) // 单个字符是回文
            f[i][j] = true;
          else if (s[i] == s[j]) // 判断两端字符是否相等
            if (i + 1 > j - 1 || f[i + 1][j - 1]) // 内部子串为空或回文
              f[i][j] = true;
        }

      // 开始回溯搜索
      dfs(s, 0);
      return res;
    }

    void dfs(std::string s, int u)
    {
      // 如果已经分割到字符串末尾，保存当前路径
      if (u == s.size())
      {
        res.push_back(path);
        return;
      }

      // 遍历从 u 开始的每个子串
      for (int i = u; i < s.size(); ++i)
        if (f[u][i]) // 如果当前子串是回文
        {
          path.push_back(s.substr(u, i - u + 1)); // 加入路径
          dfs(s, i + 1); // 递归处理剩余部分
          path.pop_back(); // 回溯
        }
    }
};