- 戳气球
有 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
题解:
先将原始数组首尾两端都插入一个数字1,为了符合题目要求做的处理。
于是样例就变成了[1,3,1,5,8,1],那么题目就变成了在[0,5]之间找到一个数字nums[x],然后再对[0,x]和[x,5]分别处理,一直递归,于是有了如下递归解法:
int solve(int l,int r,vector<int>& nums)
{
if(l==r)
return nums[l];
int ans=0;
for(int i=l+1;i<r;i++)
{
ans=max(ans,solve(l,i,nums)+solve(i,r,nums)+nums[l]*nums[i]*nums[r]);
}
return ans;
}
这样就比较容易理解了,但是这样的递归明显超时,因为存在很多重复运算,于是我们用数组将结果保存下来,有了如下最终代码。
AC代码:
class Solution {
public:
int dp[505][505];表示区间[i,j]能得到的最大结果
int maxCoins(vector<int>& nums) {
nums.insert(nums.begin(),1);
nums.push_back(1);
memset(dp,0,sizeof(dp));
for(int i=0;i<nums.size();i++)
dp[i][i]=nums[i];
for(int j=1;j<nums.size();j++)
{
for(int i=j-1;i>=0;i--)
{
for(int k=i+1;k<j;k++)
{
dp[i][j]=max(dp[i][j],dp[i][k]+dp[k][j]+nums[i]*nums[k]*nums[j]);
}
}
}
return dp[0][nums.size()-1];
}
};