【Leetcode】1278. Palindrome Partitioning III

题目地址：

https://leetcode.com/problems/palindrome-partitioning-iii/description/

给定一个长$n$的字符串$s$，再给定一个不超过$n$的正整数$k$，允许将$s$划分为$k$个非空子串，每个子串修改若干个字符使得这$k$个子串都变成回文串。问被修改的位置的个数的最小值。

设$f[n][k]$是长$n$的前缀划分为$k$个非空子串，至少修改多少个字符能使得每一部分都成为回文串，那么可以按照最后一段是哪一段来分类，如果最后一段长度是$a$，那么此时最少修改次数就是$f[n-a][k-1]+c[n-a][n-1]$，其中$c[i][j]$是$s[i:j]$修改多少次可以成为回文串。$c$可以预处理得到。代码如下：

class Solution {
 public:
  int palindromePartition(string s, int k) {
    int n = s.size();
    int cost[n][n];
    memset(cost, 0, sizeof cost);
    for (int len = 2; len <= n; len++)
      for (int l = 0; l + len - 1 < n; l++) {
        int r = l + len - 1;
        cost[l][r] = (s[l] == s[r] ? 0 : 1);
        if (len > 2) cost[l][r] += cost[l + 1][r - 1];
      }

    int f[n + 1][k + 1];
    memset(f, 0x3f, sizeof f);
    f[0][0] = 1;
    for (int i = 1; i <= n; i++) {
      f[i][1] = cost[0][i - 1];
      for (int j = 2; j <= k; j++)
        for (int l = 1; l <= i; l++)
          f[i][j] = min(f[i][j], f[i - l][j - 1] + cost[i - l][i - 1]);
    }
    return f[n][k];
  }
};

时空复杂度$O(nk)$。