题目描述:
给出一些不同颜色的盒子,盒子的颜色由数字表示,即不同的数字表示不同的颜色。
你将经过若干轮操作去去掉盒子,直到所有的盒子都去掉为止。每一轮你可以移除具有相同颜色的连续 k 个盒子(k >= 1),这样一轮之后你将得到 k*k 个积分。
当你将所有盒子都去掉之后,求你能获得的最大积分和。
思路:
动态规划
设dp[l][r][k]表示以下标l开始,以小下标r结束,k表示在下标r后面紧接着有k个元素值与boxes[r]相等,的最大积分和
比如:
代码如下:
class Solution {
public:
int removeBoxes(vector<int>& boxes) {
vector<vector<vector<int>>>dp(100,vector<vector<int>>(100,vector<int>(100,0)));//
return get_dp(boxes,0,boxes.size()-1,0,dp);
}
int get_dp(vector<int>boxes,int l,int r,int k,vector<vector<vector<int>>>dp){
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]=get_dp(boxes,l,r-1,0,dp)+(k+1)*(k+1);
for(int i=l;i<r;i++){
if(boxes[i]==boxes[r])
dp[l][r][k]=max(dp[l][r][k],get_dp(boxes,i+1,r-1,0,dp)+get_dp(boxes,l,i,k+1,dp));
}
return dp[l][r][k];
}
};