Leetcode 546. 移除盒子 C++

Leetcode 546. 移除盒子

题目

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

示例：

输入:

[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 分)

**提示：**盒子的总数 n 不会超过 100。

题解

记忆化深搜dfs、动态规划
dp[i][j][k]表示从i到j,且j之后有k个元素与boxes[j]相同元素的最大分数,
那么我们获得最大分数就为 dp[i][j-1][0]+(k+1)*(k+1)
同时我们还需要考虑，i与j之间如果有和boxes[j]相同的元素，那么我们就需要考虑这种情况，假设i···x···j，其中boxes[x]==boxes[j],则 dp[i][j][k] = dp[i][x][k+1] + dp[x+1][j-1][0];
详细过程见代码

代码

	int maxGrade(int l,int r,int k,vector<vector<vector<int>>>& dp,vector<int>& boxes){
        if(l > r)   return 0;
        if(dp[l][r][k] != 0)    return dp[l][r][k];
        while(r>l && boxes[r]==boxes[r-1]){		//保证最后k+1个元素颜色相等
            r--;
            k++;
        }
        dp[l][r][k] = maxGrade(l,r-1,0,dp,boxes) + (k+1)*(k+1);
        for(int i=l; i<r; i++){		//考虑先去掉中间几个和boxes[r]颜色不相等的盒子，这样最后颜色相等的盒子会变多
            if(boxes[i] == boxes[r]){
                dp[l][r][k] = max(dp[l][r][k],maxGrade(l,i,k+1,dp,boxes)+maxGrade(i+1,r-1,0,dp,boxes));
            }
        }
        return dp[l][r][k];
    }
    int removeBoxes(vector<int>& boxes) {
        int n = boxes.size();
        vector<vector<vector<int>>> dp(n,vector<vector<int>>(n,vector<int>(n+1,0)));
        return maxGrade(0,n-1,0,dp,boxes);
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/remove-boxes
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。