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

思路

• 如果数字中的数字为3, 1, 5, 8。先计算分别先戳破每一个气球得到的乘积数。
• 然后对于3和1。如果后戳破3，那么先戳破1。如果后戳破1，则后戳破3。同理1和5， 5和8。
• 对于3,1,5。如果最后破3，那么就先破1和5，由第二步得到。如果最后破1，先破3和5。如果最后破5，先破3和1。1,5,8同理。
• 对于3,1,5,8。最后破3，先破1,5,8。最后破1，破5，破8同理。

代码

class Solution {
public:
    int maxCoins(vector<int>& nums) {
        int n = nums.size();
        //vector<int> newNums( n + 2, 1 );
        int newNums[n+2];
        memset( newNums, 0, sizeof(newNums) );
        newNums[0] = newNums[n+1] = 1;

        //vector<vector<int>> dp( n+2, vector<int>( n+2, 0) );
        int dp[n+2][n+2];
        memset( dp, 0, sizeof(dp) );

        for ( int i = 1; i <= n; ++i ) 
            newNums[i] = nums[i-1];

        for ( int step = 0; step < n; ++step ) {
            for ( int start = 1; start + step <= n; ++start ) {
                int end = start + step;

                for ( int k = start; k <= end; ++k ) {
                    int temp = dp[start][k-1] + dp[k+1][end]; 
                    temp += newNums[k] * newNums[start-1] * newNums[end+1];
                    dp[start][end] = max( dp[start][end], temp );
                }
            }
        }

        return dp[1][n];
    }
};