leetcode 312. 戳气球

@(labuladong的算法小抄)[dp]

leetcode 312. 戳气球

题目描述

解题思路

参考：labuladong的算法小抄P181

class Solution {
    public int maxCoins(int[] nums) {
        int n = nums.length;
        if (n == 0) return 0;
        /* 在两端加入两个虚拟气球 */
        int[] points = new int[n + 2];
        points[0] = points[n + 1] = 1;
        for (int i = 0; i < n; i++)
            points[i + 1] = nums[i];
        /* 定义：戳破气球i和气球j之间（开区间，不包括i和j）的所有气球，可以获得的最高分数为dp[i][j] */
        /* base case ： if i >= j - 1, then dp[i][j] = 0 */
        int[][] dp = new int[n + 2][n + 2];
        /* i从下往上 */
        for (int i = n - 1; i >= 0; i--) {
            /* j从左往右 */
            for (int j = i + 2; j <= n + 1; j++) {
                /* 做选择，k是最后戳破的气球 */
                for (int k = i + 1; k <= j - 1; k++) {
                    dp[i][j] = Math.max(
                            dp[i][j],
                            dp[i][k] + dp[k][j] + points[i] * points[k] * points[j]
                    );
                }
            }
        }
        
        return dp[0][n + 1];
    }
}