312. Burst Balloons

题目

题意

炸掉i处的气球能获得$nums[i-1]*nums[i]*nums[i+1]$, 用过的气球就没有了, 可以将最左端和最右端看作有一个额外的1, 求总收益的最大值

分析

参考了这里有例子分析

• $dp[left][right]$表示炸掉[left, right]之间的气球能获得的最大收益

• 依次遍历不同长度len的子串s[left, right] (left∈[0, n -len], right = left+len-1);

• 用k∈[left, right] 将s[left, right]分割成s[left, k-1], s[k+1, right];
  求在子串s[left, right]中,最后炸掉第k个气球能获得的最大收益

• 其中最后炸掉第k个气球时的收益为$nums[left-1]*nums[k]*nums[right+1]$. 注意这里左右两侧的气球的取值, 因为k是[left, right]的最后一个气球了, 所以炸掉k的时候, k的左侧为left-1, 右侧为right+1

• 转移方程:
  $dp[left][right] = max(dp[left][right], dp[left][k-1]+dp[k+1][right]+nums[left-1]*nums[k]*nums[right+1])$

实现

class Solution {
public:
    int maxCoins(vector<int>& nums) {
        int n = nums.size();
        nums.push_back(1);
        nums.insert(nums.begin(), 1);

        vector<vector<int> > dp(nums.size(), vector<int>(nums.size(), 0));
        for (int len = 1; len <= n; len++) {
            for (int left = 1; left <= n-len+1; left++) {
                int right = left+len-1;
                for (int k = left; k <= right; k++) {
                    dp[left][right] = max(dp[left][right], dp[left][k-1]+dp[k+1][right]+nums[left-1]*nums[k]*nums[right+1]);
                }
            }
        }
        return dp[1][n];
    }
};