超详细！腾讯高频算法题详解（带leetcode答案）

腾讯面试题： 移除箱子

描述
给出几个不同颜色的箱子，用不同的正数表示。
你可能会经历几轮拆除箱子，直到没有箱子为止。 每次你可以选择一些具有相同颜色的连续箱子（由k个箱子组成，k >= 1），删除它们并得到k * k分。
找到您可以获得的最高分。
箱子n的数量不会超过100。
在线评测地址

样例 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] (3*3=9分) 
----> [1, 3, 3, 3, 1] (1*1=1分) 
----> [1, 1] (3*3=9分) 
----> [] (2*2=4分)

样例 2:

输入：[1,2,3,4,5,6,7,8,9,10]
输出：10
解释：
每轮得到1*1分，一共10轮。

解题思路
考点：区间dp
题解：

• scoresik 表示的是通过移除boxesi,j中的箱子，且此时在boxesi前面有k个与boxesi颜色相同的箱子。因此，对于第i个箱子，如果将其和前面的k个箱子一起移除，那么此时可以获得的分数，可以表示为： scoresik = (1 + k) * (1 + k);
• 同时对于第i个箱子，还有其他的方案来移除，即可以将boxesi,j中的某一个箱子一起移除，这个箱子可以表示为boxesm，此时boxesm == boxesi。此时可以获得的分数，可以表示为： result = max(result, scoresi + 10 + scoresmk + 1);
  源代码

class Solution:
    """
    @param boxes: List[int]
    @return: return an integer
    """
    def removeBoxes(self, boxes):
        # write your code here
        N = len(boxes)
        memo = [[[0]*N for _ in xrange(N) ] for _ in xrange(N) ]

        def dp(i, j, k):
            if i > j: return 0
            if not memo[i][j][k]:
                m = i
                while m+1 <= j and boxes[m+1] == boxes[i]:        
                    m += 1
                i, k = m, k + m - i        
                ans = dp(i+1, j, 0) + (k+1) ** 2        #i位置箱子和前面k个一起移除
                for m in xrange(i+1, j+1):
                    if boxes[i] == boxes[m]:        #如果和i处的相等，可以一起移除
                        ans = max(ans, dp(i+1, m-1, 0) + dp(m, j, k+1))        #移除i和m的箱子需要先移除[i+1,m-1]内的箱子，i左面有k个箱子，m左面就有k+1个
                memo[i][j][k] = ans
            return memo[i][j][k]

        return dp(0, N-1, 0)

更多题解参考