LeetCode 312. 戳气球

题目描述

有 n 个气球，编号为0 到 n-1，每个气球上都标有一个数字，这些数字存在数组 nums 中。

现在要求你戳破所有的气球。每当你戳破一个气球 i 时，你可以获得 nums[left] * nums[i] * nums[right] 个硬币。 这里的 left 和 right 代表和 i 相邻的两个气球的序号。注意当你戳破了气球 i 后，气球 left 和气球 right 就变成了相邻的气球。

求所能获得硬币的最大数量。

说明:

• 你可以假设 nums[-1] = nums[n] = 1，但注意它们不是真实存在的所以并不能被戳破。
• 0 ≤ n ≤ 500, 0 ≤ nums[i] ≤ 100

示例:

输入: [3,1,5,8]
输出: 167 
解释: nums = [3,1,5,8] --> [3,5,8] -->   [3,8]   -->  [8]  --> []
     coins =  3*1*5      +  3*5*8    +  1*3*8      + 1*8*1   = 167

问题分析

此题用动态规划来解决，创建一个二维dp数组，dp[i][j]表示在数组nums的从 i 到 j 范围内，可以获得最大硬币的数量。为了计算戳破边缘的气球的硬币数，我们先在nums数组的首尾各加上元素1，然后我们创建的dp数组大小是n + 2乘n + 2的，初始值为0，但我们戳破气球的范围是从1到n的，不包括0和n+1，所以计算dp数组时也是从dp[1][1]到dp[n][n]的。我们先计算区间长度len为1所对应的dp值，然后是2，3，……，一直到n。在计算dp[i][j]时，对应的区间长度len为j - i + 1，我们用变量k遍历这个区间中的位置，先戳破从i到k-1的气球，获得了dp[i][k-1]的硬币，然后戳破从k+1到 j 的气球，获得了dp[k+1][j]的硬币，最后戳破位置为k的气球，可获得nums[i-1]*nums[k]*nums[j+1]的硬币，把三者加起来，就是在这个区间中最后戳破位置为k的气球所能获得的最大硬币数，我们把k的位置在 i 到 j 中遍历一遍，所得到的最大值就是dp[i][j]。如此进行，最后返回dp[1][n]。

代码实现

class Solution {
public:
    int maxCoins(vector<int>& nums) {
        int n = nums.size();
        nums.insert(nums.begin(), 1);
        nums.push_back(1);
        vector<vector<int>> dp(n + 2, vector<int>(n + 2, 0));
        for(int len = 1; len <= n; len++){
            for(int i = 1; i <= n - len + 1; i++){
                int j = i + len - 1;
                for(int k = i; k <= j; k++){
                    dp[i][j] = max(dp[i][j], dp[i][k-1] + dp[k+1][j] + nums[i-1] * nums[k] * nums[j+1]);
                }
            }
        }
        return dp[1][n];
    }
};