有 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 = 3*1*5 + 3*5*8 + 1*3*8 + 1*8*1 = 167
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/burst-balloons
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题解、动态规划,定义dp[i][j]为戳i,j索引之间气球可得到的最多硬币数量,k为最后一个戳的索引。防止索引越界,左右各添加一个元素1。
class Solution{
public int maxCoins(int[] nums){
int n = nums.length;
int[] val = new int[n + 2];
val[0] = val[n + 1] = 1;
System.arraycopy(nums, 0, val, 1, n);
int[][] dp = new int[n + 2][n + 2];
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++){
int sum = val[i] * val[j] * val[k];
sum += dp[i][k] + dp[k][j];
dp[i][j] = Math.max(sum, dp[i][j]);
}
}
}
return dp[0][n + 1];
}
}