每日一题算法：2020年8月15日 [移除盒子] removeBoxes

2020年8月15日 移除盒子 removeBoxes

class Solution {
    public int removeBoxes(int[] boxes) {

    }
}

解题思路：

看完题干，我只有一个问题，困难题就这？这也能算是困难题了吗？我觉得甚至没有昨天的简单题有难度，先说我的思路，遍历数组取出所有颜色的个数，去重，存在哈希表中，哈希表的键是key，值是个数。看到下面说长度不会超过100，那么改用数组代替哈希表。

代码实现：

    public int removeBoxes(int[] boxes) {
        int[] array=new int[100];
        for (int i=0;i<boxes.length;i++){
            array[boxes[i]]++;
        }
        int res=0;
        for (int i=0;i<100;i++){
            res+=array[i]*array[i];
        }
        return res;
    }

果然没有那么简单，再次确认了题目之后发现是只有连续的才能移除。

方法2，递归+剪枝。

我们可以使用递归，遍历拆除每一个连续部分后，构建新的数组用于递归，，直到数组为空为止，但是我觉得这样子大概率会导致超时，虽然我们可以使用数组复制的方法来赋值数组。所以我们还需要 一个剪枝的算法来规避掉一些重复的计算。

下面是没有使用优化的代码。

public int removeBoxes(int[] boxes) {
    return removeBoxes(boxes,0 );
}
public int removeBoxes(int[] boxes,int val) {

    int len=boxes.length;
    if (len==0)
        return 0;
    int flag=boxes[0];
    int num=0;
    int res=0;
    for (int i=0;i<len;i++){
        //如果连续了，那么长度+1
        if (boxes[i]==flag){
            num++;
        }
        //否则的话，长度清零并且递归
        else {
            //将其与部分复制为数组
            int[] newarray=new int[len-num];
            System.arraycopy(boxes,0 ,newarray ,0 ,i-num);
            System.arraycopy(boxes,i,newarray ,i-num ,len-i);

            res=Integer.max(removeBoxes(newarray,val+num*num),res );

            num=1;
            flag=boxes[i];

        }
    }
    return Integer.max(res,val+num*num);
}

没有想出优化的方式，看答案：

class Solution {
    public int removeBoxes(int[] boxes) {
        int[][][] dp = new int[100][100][100];
        return calculatePoints(boxes, dp, 0, boxes.length - 1, 0);
    }

    public int calculatePoints(int[] boxes, int[][][] dp, int l, int r, int k) {
        if (l > r) return 0;
        if (dp[l][r][k] != 0) return dp[l][r][k];
        while (r > l && boxes[r] == boxes[r - 1]) {
            r--;
            k++;
        }
        dp[l][r][k] = calculatePoints(boxes, dp, l, r - 1, 0) + (k + 1) * (k + 1);
        for (int i = l; i < r; i++) {
            if (boxes[i] == boxes[r]) {
                dp[l][r][k] = Math.max(dp[l][r][k], calculatePoints(boxes, dp, l, i, k + 1) + calculatePoints(boxes, dp, i + 1, r - 1, 0));
            }
        }
        return dp[l][r][k];
    }
}

作者：LeetCode-Solution
链接：https://leetcode-cn.com/problems/remove-boxes/solution/yi-chu-he-zi-by-leetcode-solution/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。