LeetCode 546 移除盒子

题目描述:
给出一些不同颜色的盒子,盒子的颜色由数字表示,即不同的数字表示不同的颜色。
你将经过若干轮操作去去掉盒子,直到所有的盒子都去掉为止。每一轮你可以移除具有相同颜色的连续 k 个盒子(k >= 1),这样一轮之后你将得到 k*k 个积分。
当你将所有盒子都去掉之后,求你能获得的最大积分和。

思路:
动态规划
设dp[l][r][k]表示以下标l开始,以小下标r结束,k表示在下标r后面紧接着有k个元素值与boxes[r]相等,的最大积分和
比如:
LeetCode 546 移除盒子_第1张图片
代码如下:

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];
    }
};

你可能感兴趣的:(leetcode,leetcode,算法,动态规划,数据结构,c++)