LeetCode 312. 戳气球--区间DP

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

题解：

先将原始数组首尾两端都插入一个数字1，为了符合题目要求做的处理。
于是样例就变成了[1,3,1,5,8,1]，那么题目就变成了在[0,5]之间找到一个数字nums[x]，然后再对[0,x]和[x,5]分别处理，一直递归，于是有了如下递归解法：

    int solve(int l,int r,vector<int>& nums)
    {
        if(l==r)
        return nums[l];
        int ans=0;
        for(int i=l+1;i<r;i++)
        {
            ans=max(ans,solve(l,i,nums)+solve(i,r,nums)+nums[l]*nums[i]*nums[r]);
        }
        return ans;
    }

这样就比较容易理解了，但是这样的递归明显超时，因为存在很多重复运算，于是我们用数组将结果保存下来，有了如下最终代码。

AC代码：

class Solution {
public:
    int dp[505][505];表示区间[i,j]能得到的最大结果
    int maxCoins(vector<int>& nums) {
        nums.insert(nums.begin(),1);
        nums.push_back(1);
        memset(dp,0,sizeof(dp));
        for(int i=0;i<nums.size();i++)
        dp[i][i]=nums[i];
        for(int j=1;j<nums.size();j++)
        {
            for(int i=j-1;i>=0;i--)
            {
                for(int k=i+1;k<j;k++)
                {
                    dp[i][j]=max(dp[i][j],dp[i][k]+dp[k][j]+nums[i]*nums[k]*nums[j]);
                }
            }
        }
        return dp[0][nums.size()-1];
    }
};