546. 移除盒子

Problem: 546. 移除盒子

思路

这是一个典型的动态规划问题。我们使用三维数组 dp[l][r][k] 表示从索引 l 到 r 范围内，最后一个盒子颜色与 boxes[l] 相同的盒子数目为 k 时，所能获得的最大分数。这里 k 实际上是盒子 boxes[l] 的连续重复次数。状态转移的关键在于找到能够与 boxes[l] 形成匹配的盒子，从而增加分数并更新状态。

解题方法

初始化一个三维数组 dp，用于存储子问题的答案。
使用递归函数 f 来计算从索引 l 到 r 的最大得分，其中 k 是当前连续相同颜色盒子的数量。在 f 函数中，首先检查是否已经计算过该状态，如果已知答案则直接返回。接下来寻找与 boxes[l] 相同颜色的下一个盒子，并计算连续相同颜色盒子的数量。根据找到的连续盒子数量，计算移除这些盒子的分数。对于 l 和 r 范围内的每一个可能位置 m，检查是否有盒子与 boxes[l] 颜色相同且可形成匹配，以求得最优解。最终返回该状态下的最大得分。

复杂度

时间复杂度:

$O(n^4)$，其中 n 是盒子的数量。这是因为对于每一对 (l, r)，我们需要遍历所有 m 的可能值。

空间复杂度:

$O(n^3)$，用于存储动态规划的状态数组。

Code

class Solution {
    public int removeBoxes(int[] boxes) {
        int n = boxes.length;
        int[][][] dp = new int[n][n][n];
        return f(boxes, 0, n - 1, 0, dp);
    }
    public int f(int[] boxes, int l, int r, int k, int[][][] dp) {
        if(l > r) {
            return 0;
        }
        if(dp[l][r][k] > 0) {
            return dp[l][r][k];
        }
        int s = l;
        while(s + 1 <= r && boxes[s] == boxes[s + 1]) {
            s++;
        }
        int cnt = k + s - l + 1;
        int ans = cnt * cnt + f(boxes, s + 1, r, 0, dp);
        for(int m = s + 2; m <= r; m++) {
            if(boxes[l] == boxes[m] && boxes[m - 1] != boxes[m]) {
                ans = Math.max(ans, f(boxes, s + 1, m - 1, 0, dp) + f(boxes, m, r, cnt, dp));
            }
        }
        dp[l][r][k] = ans;
        return ans;


    }
}