移除盒子

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

示例 1：
输入:

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

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

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

class Solution {
private:
    int mem[101][101][101];
public:
    int removeBoxes(vector<int>& boxes) {
        int len = boxes.size();
        memset(mem,0,sizeof(mem));
        return dfs(boxes,0,len-1,0);
    }
    int dfs(vector<int>& boxes,int l,int r,int k){
        if(r<l){
            return 0;
        }
        while(l<r&&boxes[r]==boxes[r-1]){
            r--;
            k++;
        }
        if(mem[l][r][k]>0){
            return mem[l][r][k];
        }
        int& ans = mem[l][r][k];
        ans = dfs(boxes,l,r-1,0)+(k+1)*(k+1);
        for(int i=l;i<r;i++){
            if(boxes[i]==boxes[r]){
                ans = max(ans, dfs(boxes,l,i,k+1)+dfs(boxes,i+1,r-1,0));
            }
        }
        return ans;

    }
};