移除盒子

一、题目描述

题目链接：https://leetcode-cn.com/problems/remove-boxes/

 

二、题目分析

本题使用到三维dp，参考题解：https://leetcode-cn.com/problems/remove-boxes/solution/guan-fang-fang-fa-2ji-yi-hua-sou-suo-dong-hua-tu-j/

配合注释食用更佳

 

三、代码

    public int removeBoxes(int[] boxes) {
        // 矩阵对应到函数calculatePoints中为dp[left][right][k],left和right代表boxes的起点和终点
        // k代表在这串序列后有k个数与boxes[right]相同
        int[][][] dp = new int[100][100][100];
        return calculatePoints(boxes,dp,0,boxes.length-1,0);
    }

    // 此函数用于计算left到right的boxes的dp最大值
    private int calculatePoints(int[] boxes, int[][][] dp, int left, int right, int k) {
        if (left > right) return 0;
        if (dp[left][right][k] != 0) return dp[left][right][k];
        //更正right与k
        while (right > left && boxes[right] == boxes[right-1]) {
            right --;
            k ++;
        }
        //对应题解中策略1,直接将后缀的相同数字去除，有(k+1)*(k+1)的得分(+1是因为right也要算进去)
        //后面的就算算完了,再算前面的,即calculatePoints(boxes,dp,left,right-1,0)
        dp[left][right][k] = calculatePoints(boxes,dp,left,right-1,0) + (k+1)*(k+1);
        //对应题解策略2
        for (int i = left; i < right; i ++ ) {
            if (boxes[i] == boxes[right]) {
                //i与right中间的就是题解中说的杂鱼数字,计算杂鱼数字序列dp为calculatePoints(boxes,dp,i+1,right-1,0)
                //剩下的部分就相当于没有杂鱼数字了,对应就是left到i,后面紧跟k+1个相同的数字
                dp[left][right][k] = Math.max(dp[left][right][k], calculatePoints(boxes,dp,left,i,k+1) + calculatePoints(boxes,dp,i+1,right-1,0));
            }
        }
        return dp[left][right][k];
    }