Problem P27. [算法课动态规划] 戳气球
有 n 个气球,编号为0 到 n - 1,每个气球上都标有一个数字,这些数字存在数组 nums 中。
现在要求你戳破所有的气球。戳破第 i 个气球,你可以获得 nums[i−1]×nums[i]×nums[i+1] 枚硬币。 这里的 i - 1 和 i + 1 代表和 i 相邻的两个气球的序号。如果 i - 1或 i + 1 超出了数组的边界,那么就当它是一个数字为 1 的气球。
求所能获得硬币的最大数量
提示:
- n == nums.length
- 1 <= n <= 300
- 0 <= nums[i] <= 100
- 题目数据保证运算过程不超过 int 所能表示的范围
输入
输入一行数组num
输出
输出所能获得硬币的最大数量
样例
标准输入复制文本 |
3 1 5 8 |
标准输出复制文本 |
167 |
标准输入复制文本 |
1 5 |
标准输出复制文本 |
10 |
#include<iostream>
#include<vector>
using namespace std;
class Solution {
public:
int rob(vector<int>& nums)
{
nums.push_back(1);
nums.emplace(nums.begin(), 1);
int dp[nums.size()][nums.size()];
for(int start=nums.size()-1;start>=0;start--)
{
for(int end=start;end<nums.size();end++)
{
int max = 0;
for(int k=start+1;k<end;k++)
{
int temp = dp[start][k]+dp[k][end]+nums[start]*nums[k]*nums[end];
if(temp>max)
max = temp;
}
dp[start][end]= max;
}
}
return dp[0][nums.size()-1];
}
};
int main()
{
Solution sol;
vector<int> nums;
vector<vector<int>> rec;
int i;
while (cin >> i)
{
nums.push_back(i);
if (cin.get() == '\n') break;
}
cout << sol.rob(nums)<<endl;
return 0;
}