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
示例:
输入: [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
解题
最大值——穷举——回溯 or 动态规划(记忆穷举);
推导——先选最后一个被戳破的气球;
设置dp[i][j]作为i和j中戳破气球得到的最大硬币数;
k为i,j中最后一个被戳破的气球;
则dp[i][j]=max(dp[i][j],dp[i][k]+dp[k][j]+bal[i]*bal[j]*bal[k]);
dp【i】【k】为k之前已被戳破的气球的最大和;
dp【k】【j】为k~j中已被戳破的最大和;
其加上最后被戳破的k的硬币和即为总的最大和;
推导方向——i从大到小,j从i到大,k从i到j;
由中心往两边递推,最后返回dp[0][n+1]即可;
注意点
0和n+1都为1;
class Solution {
public:
int maxCoins(vector<int>& nums) {
int n=nums.size();
ballon.resize(n+2,0);
ballon[0]=1;
ballon[n+1]=1;
for(int i=0;i<n;i++) ballon[i+1]=nums[i];
dp.resize(n+2,vector<int> (n+2,0));
for(int i=n;i>=0;i--)
for(int j=i+1;j<n+2;j++)
for(int k=i+1;k<j;k++){
dp[i][j]=max(dp[i][j],dp[i][k]+dp[k][j]+ballon[k]*ballon[i]*ballon[j]);
//选k作为最后一个戳破的
}
return dp[0][n+1];
}
private:
vector<vector<int>> dp;
vector<int>ballon;
};