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

示例1
输入：
[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. 由于戳破第一个（下标为0）的气球和最后一个（下标为n-1）的球时，相邻的气球（下标为-1和下标为n）超过数组下标
2. 所以新创建一个数组，把下标为-1和下标为n的加进来值为1
3. 正面戳破气球不好做，我们倒过来把戳破气球看成每次加一个气球。
4. 用res[i][j]代表从第i个元素到第j个元素的内的位置全部填满能够得到的最大值，注意开区间！

方法一：分治
函数f：

1. 当i+1==j时 若i=1，j=2，两者之间不能放入新的元素，返回0
2. 当res[i][j]有值时，返回（记忆化搜索）
3. 当i+1<j时,即(i,j)之间至少可以放入一个元素，放一个元素（下标为k，i<k<j）就把区间拆分成两个区间，再次调用f函数，取区间最大值
4. 放一个元素，可以得到vals[i]*vals[k]*vals[j],区间变成（i,k),(k,j),这两个区间在调用f函数，(i,j)根据不同的k就是切分区间，取最大值
5. 举例

    nums： 3 1 5 8
    val：1 3 1 5 8 1
    1.当i=0,j=1
        中间插不了值，返回0
    2.当i=0，j=2
        中间可以插入唯一的1，返回1*3*1
    3.当i = 0 j = 3
        k=1时，res(0,3) = vals[i]*vals[k]*vals[j] = 1*3*5+f(0,1)+f(1,3) = 15+0+3*1*5 = 30
        k=2时，res(0,3) = vals[i]*vals[k]*vals[j] = 1*1*5+f(0,2)+f(2,3) = 5+1*3*1+0 = 8
        res(0,3) = max(k=1时的值，k=2时的值）= 30

方法二：动态规划

1. 分治主要是自顶向下的思路，动态规划主要是自低向上，前面的都差不多
2. 但是转移的顺序比较关键！！！要倒着来，从小到大

代码：
方法一：分治

#include <stdio.h>
#include <iostream>
#include <vector>
using namespace std;
class Solution {
    vector<int> val;
    vector<vector<int>> res;
public:
     int f(int first,int end){
        if(first+1==end)//区间内放不进气球
            return 0;
        if(res[first][end]!=-1)//记忆搜索，有值更新过才能用
            return res[first][end];
         for(int i=first+1;i<end;i++){
             int tmp = val[first]*val[i]*val[end];
             tmp += f(first,i)+f(i,end);//递归调用
             res[first][end] = max(res[first][end],tmp);//取最大值
         }
         return res[first][end];
     }
     int maxCoins(vector<int>& nums) {
         int n = nums.size();
         val.resize(n+2);
         for(int i=1;i<=n;i++)
            val[i] = nums[i-1];
         val[0]=val[n+1]=1;//创建新数组
         res.resize(n+2,vector<int>(n+2,-1));
         return f(0,n+1);
     }
 };
 int main()
{
    vector<int> v(4,0);
    v[0]=3;v[1]=1;v[2]=5;v[3]=8;
    Solution s;
    cout<<s.maxCoins(v);
    return 0；
}

方法二：动态规划

#include <stdio.h>
#include <iostream>
#include <vector>
using namespace std;
class Solution {
public:
    int maxCoins(vector<int>& nums) {
        int n = nums.size();
        vector<vector<int>> res(n + 2, vector<int>(n + 2));
        vector<int> val(n + 2);
        for (int i = 1; i <= n; i++) {
            val[i] = nums[i - 1];
        }
        val[0] = val[n + 1] = 1;
        //更新后才能用值，和背包问题类似
        for (int i = n - 1; i >= 0; i--) {
            for (int j = i + 2; j <= n + 1; j++) {
                for (int k = i + 1; k < j; k++) {//即(i,j)之间选择一个元素
                    int sum = val[i] * val[k] * val[j];
                    sum += res[i][k] + res[k][j];//找区间的值
                    res[i][j] = max(res[i][j], sum);
                }
            }
        }
        return res[0][n + 1];
    }
};
int main()
{
    vector<int> v(4,0);
    v[0]=3;v[1]=1;v[2]=5;v[3]=8;
    Solution s;
    cout<<s.maxCoins(v);
    return 0;
}

今天也是爱zz的一天哦！