有 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
思路与代码
动态规划
dp[i][j]表示戳破i和j之间的气球能获得最大数量的金币,不含i、j
class Solution {
public int maxCoins(int[] nums) {
int n = nums.length;
int[] new_nums = new int[n+2];
new_nums[0] = 1;
new_nums[n+1] = 1;
System.arraycopy(nums,0,new_nums,1,n);
int[][] dp = new int[n + 2][n + 2];
for (int interval = 2; interval <= n + 1; interval++) {
for (int start = 0; start + interval <= n + 1; start++) {
for (int k = start + 1; k < start + interval; k++) {
dp[start][start + interval] = Math.max(dp[start][start + interval],
new_nums[start]*new_nums[k]*new_nums[start+interval] + dp[start][k] + dp[k][start + interval]) ;
}
}
}
return dp[0][n+1];
}
}