题目
有 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
示例1
输入:
[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
解法
- 由于戳破第一个(下标为0)的气球和最后一个(下标为n-1)的球时,相邻的气球(下标为-1和下标为n)超过数组下标
- 所以新创建一个数组,把下标为-1和下标为n的加进来值为1
- 正面戳破气球不好做,我们倒过来把戳破气球看成每次加一个气球。
- 用res[i][j]代表从第i个元素到第j个元素的内的位置全部填满能够得到的最大值,注意开区间!
方法一:分治
函数f:
- 当i+1==j时 若i=1,j=2,两者之间不能放入新的元素,返回0
- 当res[i][j]有值时,返回(记忆化搜索)
- 当i+1<j时,即(i,j)之间至少可以放入一个元素,放一个元素(下标为k,i<k<j)就把区间拆分成两个区间,再次调用f函数,取区间最大值
- 放一个元素,可以得到vals[i]*vals[k]*vals[j],区间变成(i,k),(k,j),这两个区间在调用f函数,(i,j)根据不同的k就是切分区间,取最大值
- 举例
nums: 3 1 5 8
val:1 3 1 5 8 1
1.当i=0,j=1
中间插不了值,返回0
2.当i=0,j=2
中间可以插入唯一的1,返回1*3*1
3.当i = 0 j = 3
k=1时,res(0,3) = vals[i]*vals[k]*vals[j] = 1*3*5+f(0,1)+f(1,3) = 15+0+3*1*5 = 30
k=2时,res(0,3) = vals[i]*vals[k]*vals[j] = 1*1*5+f(0,2)+f(2,3) = 5+1*3*1+0 = 8
res(0,3) = max(k=1时的值,k=2时的值)= 30
方法二:动态规划
- 分治主要是自顶向下的思路,动态规划主要是自低向上,前面的都差不多
- 但是转移的顺序比较关键!!!要倒着来,从小到大
代码:
方法一:分治
#include <stdio.h>
#include <iostream>
#include <vector>
using namespace std;
class Solution {
vector<int> val;
vector<vector<int>> res;
public:
int f(int first,int end){
if(first+1==end)//区间内放不进气球
return 0;
if(res[first][end]!=-1)//记忆搜索,有值更新过才能用
return res[first][end];
for(int i=first+1;i<end;i++){
int tmp = val[first]*val[i]*val[end];
tmp += f(first,i)+f(i,end);//递归调用
res[first][end] = max(res[first][end],tmp);//取最大值
}
return res[first][end];
}
int maxCoins(vector<int>& nums) {
int n = nums.size();
val.resize(n+2);
for(int i=1;i<=n;i++)
val[i] = nums[i-1];
val[0]=val[n+1]=1;//创建新数组
res.resize(n+2,vector<int>(n+2,-1));
return f(0,n+1);
}
};
int main()
{
vector<int> v(4,0);
v[0]=3;v[1]=1;v[2]=5;v[3]=8;
Solution s;
cout<<s.maxCoins(v);
return 0;
}
方法二:动态规划
#include <stdio.h>
#include <iostream>
#include <vector>
using namespace std;
class Solution {
public:
int maxCoins(vector<int>& nums) {
int n = nums.size();
vector<vector<int>> res(n + 2, vector<int>(n + 2));
vector<int> val(n + 2);
for (int i = 1; i <= n; i++) {
val[i] = nums[i - 1];
}
val[0] = val[n + 1] = 1;
//更新后才能用值,和背包问题类似
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++) {//即(i,j)之间选择一个元素
int sum = val[i] * val[k] * val[j];
sum += res[i][k] + res[k][j];//找区间的值
res[i][j] = max(res[i][j], sum);
}
}
}
return res[0][n + 1];
}
};
int main()
{
vector<int> v(4,0);
v[0]=3;v[1]=1;v[2]=5;v[3]=8;
Solution s;
cout<<s.maxCoins(v);
return 0;
}
今天也是爱zz的一天哦!