[菜鸟训练]312. 戳气球

题目描述：

有 n 个气球，编号为0 到 n - 1，每个气球上都标有一个数字，这些数字存在数组 nums 中。
现在要求你戳破所有的气球。戳破第 i 个气球，你可以获得 nums[i - 1] * nums[i] * nums[i + 1] 枚硬币。 这里的 i - 1 和 i + 1 代表和 i 相邻的两个气球的序号。如果 i - 1或 i + 1 超出了数组的边界，那么就当它是一个数字为 1 的气球。
求所能获得硬币的最大数量。

示例 1：
输入：nums = [3,1,5,8]
输出：167
解释：
nums = [3,1,5,8] --> [3,5,8] --> [3,8] --> [8] --> []
coins = 315 + 358 + 138 + 181 = 167

示例 2：
输入：nums = [1,5]
输出：10

提示：
n == nums.length
1 <= n <= 500
0 <= nums[i] <= 100

解题思路：

dp，我们根据题意明显可以使用动态规划的思想，我们假设最后戳破的气球的位置是k，k是除了区间端点内的任意点，此时该气球区间只剩下三个气球，两个区间端点和k点，在这里根据题意，如果 i - 1或 i + 1 超出了数组的边界，那么就当它是一个数字为 1 的气球。 所以我们不妨将nums数组的两端进行一下扩充，及将其两端各添加一个元素1。我们可以得到，dp[i][j] = dp[i][k] + tmp[i] * tmp[k] * tmp[i + t - 1] + dp[k][j]; 而此时k的选择会导致我们最终获得的金币数量。所以我们需要每次都去找获得金币数量最多的情况，k的取值范围是(i,j)。

代码：

public class LC312 {
    //dp，时间复杂度O(n^3) 空间复杂度O(n^2)
    public int maxCoins(int[] nums) {
        int len = nums.length;

        //辅助数组，在原数组得基础上将该数组进行扩充，即将边界值设为1
        int[] tmp = new int[len + 2];
        tmp[0] = 1;
        tmp[len + 1] = 1;

        for (int i = 0; i < len; i++){
            tmp[i + 1] = nums[i];
        }

        //造一个dp数组
        int[][] dp = new int[len + 2][ len + 2];

        //t表示区间长度
        for (int t = 3; t <= len + 2; t++){
            //i表示区间得左端点，右端点就是区间长度+起始位置-1
            for (int i = 0; i <= len + 2 - t; i++){
                int max_num = 0;
                for (int k = i + 1; k < i + t - 1; k++){
                    max_num = Math.max(max_num, dp[i][k] + tmp[i] * tmp[k] * tmp[i + t - 1] + dp[k][i + t- 1]);
                }
                dp[i][i+t-1] = max_num;
            }
        }
        //开区间
        return dp[0][len+1];
    }

    public static void main(String[] args) {
        LC312 obj = new LC312();
        System.out.println(obj.maxCoins(new int[]{3, 1, 5, 8}));
    }
}