有 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/burst-balloons
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
基本思路:dfs+记忆
- 添加边界,减少边界条件的讨论;
- 在dfs过程中,在left~right中i,表示的是最后一个戳破的气球,通过这样把大问题分化为相对独立的小问题;
- 否则dfs(left,i)会受dfs[i,right)的影响,因为戳破i后,i-1会和i+1相邻;
int dfs(int left,int right,vector<vector<int>> &memo,vector<int> &nums){
if(right<=left+1)
return 0;
if(memo[left][right]!=-1)
return memo[left][right];
for(int i=left+1;i<right;i++){
int tmp=dfs(left,i,memo,nums)+dfs(i,right,memo,nums)+nums[left]*nums[i]*nums[right];
memo[left][right]=max(tmp,memo[left][right]);
}
return memo[left][right];
}
int maxCoins(vector<int>& nums) {
int len=nums.size();
vector<vector<int>> memo(len+2,vector<int>(len+2,-1));
vector<int> new_nums(len+2);
for(int i=0;i<nums.size();i++){
new_nums[i+1]=nums[i];
}
new_nums[0]=new_nums[len+1]=1;
return dfs(0,len+1,memo,new_nums);
}
基本思路:动态规划,与dfs中的memo类似。
- dp[i][j]表示的以i和j为边界的获得的最大金币数,其中i,j是不会被戳破的
- dp[i][j]=dp[i][k]+dp[k][j]+nums[i]*nums[k]*nums[j],表示的是区间中k是最后一个被戳破的气球,故这样可以将其拆分为两个相对独立的最优子结构。
int maxCoins(vector<int>& nums) {
int len=nums.size();
vector<vector<int>> dp(len+2,vector<int>(len+2,0));
vector<int> new_nums(len+2);
for(int i=0;i<nums.size();i++){
new_nums[i+1]=nums[i];
}
new_nums[0]=new_nums[len+1]=1;
for(int j=2;j<new_nums.size();j++){
for(int i=j-2;i>=0;i--){
for(int k=i+1;k<j;k++){
dp[i][j]=max(dp[i][j],dp[i][k]+dp[k][j]+new_nums[i]*new_nums[k]*new_nums[j]);
}
}
}
return dp[0][len+1];
}