0
点赞
收藏
分享

微信扫一扫

#yyds干货盘点# LeetCode程序员面试金典:移除盒子

Resin_Wu 2024-01-14 阅读 15

题目

给出一些不同颜色的盒子 boxes ,盒子的颜色由不同的正数表示。


你将经过若干轮操作去去掉盒子,直到所有的盒子都去掉为止。每一轮你可以移除具有相同颜色的连续 k 个盒子(k >= 1),这样一轮之后你将得到 k * k 个积分。


返回 你能获得的最大积分和 。


 


示例 1:


输入:boxes = [1,3,2,2,2,3,4,3,1]

输出:23

解释:

[1, 3, 2, 2, 2, 3, 4, 3, 1]  

----> [1, 3, 3, 4, 3, 1] (3*3=9 分)  

----> [1, 3, 3, 3, 1] (1*1=1 分)  

----> [1, 1] (3*3=9 分)  

----> [] (2*2=4 分)

示例 2:


输入:boxes = [1,1,1]

输出:9

示例 3:


输入:boxes = [1]

输出:1

 

代码实现


class Solution {
    int[][][] dp;

    public int removeBoxes(int[] boxes) {
        int length = boxes.length;
        dp = new int[length][length][length];
        return calculatePoints(boxes, 0, length - 1, 0);
    }

    public int calculatePoints(int[] boxes, int l, int r, int k) {
        if (l > r) {
            return 0;
        }
        if (dp[l][r][k] == 0) {
            dp[l][r][k] = calculatePoints(boxes, l, r - 1, 0) + (k + 1) * (k + 1);
            for (int i = l; i < r; i++) {
                if (boxes[i] == boxes[r]) {
                    dp[l][r][k] = Math.max(dp[l][r][k], calculatePoints(boxes, l, i, k + 1) + calculatePoints(boxes, i + 1, r - 1, 0));
                }
            }
        }
        return dp[l][r][k];
    }
}


举报

相关推荐

0 条评论