给出一些不同颜色的盒子,盒子的颜色由数字表示,即不同的数字表示不同的颜色。
你将经过若干轮操作去去掉盒子,直到所有的盒子都去掉为止。每一轮你可以移除具有相同颜色的连续 k 个盒子(k >= 1),这样一轮之后你将得到 k*k 个积分。
当你将所有盒子都去掉之后,求你能获得的最大积分和。
这里是引用
示例:
输入: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] (33=9 分)
----> [1, 3, 3, 3, 1] (11=1 分)
----> [1, 1] (33=9 分)
----> [] (22=4 分)
提示:
1 <= boxes.length <= 100
1 <= boxes[i] <= 100
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-boxes
【动规】
来自官方题解
用 f(l, r, k) 表示移除区间[l,r] 加上该区间右边等于boxes[r] 的 k 个元素组成的这个序列的最大积分。
比如:
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/remove-boxes/solution/yi-chu-he-zi-by-leetcode-solution/
public int removeBoxes(int[] boxes) {
if (boxes.length == 0) {
return 0;
}
if (boxes.length == 1) {
return 1;
}
int[][][] dp = new int[100][100][100];
return calculatePoints(boxes, dp, 0, boxes.length - 1, 0);
}
public int calculatePoints(int[] boxes, int[][][] dp, int l, int r, int k) {
if (l > r) return 0;
if (dp[l][r][k] != 0) return dp[l][r][k];
while (l < r && boxes[r] == boxes[r - 1]) {
r--;
k++;
}
dp[l][r][k] = calculatePoints(boxes, dp, 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, dp,l,i,k + 1) + calculatePoints(boxes, dp,i + 1,r - 1,0));
}
}
return dp[l][r][k];
}