**
题目:戳气球
**
有 n 个气球,编号为0 到 n - 1,每个气球上都标有一个数字,这些数字存在数组 nums 中。
现在要求你戳破所有的气球。戳破第 i 个气球,你可以获得
nums[i−1]×nums[i]×nums[i+1] 枚硬币。 这里的 i - 1 和 i + 1 代表和 i 相邻的两个气球的序号。如果 i - 1或 i + 1 超出了数组的边界,那么就当它是一个数字为 1 的气球。
求所能获得硬币的最大数量。
输入:输入一行数组num
输出:能获得的最大硬币数
分析
参考分析链接:戳气球
由于,如果i-1或者i+1超出了数组的边界,那么就认为它是一个数字为1的气球,那么,可以假设数组的 nums[-1] = nums[n] = 1。
再确定状态:dp[i][j] = x 表示,戳破气球 i 和气球 j 之间(开区间,不包括 i 和 j)的所有气球,可以获得的最高分数为 x。
确定状态转移方程:
代码
#include<iostream>
#include<vector>
using namespace std;
class Solution {
public:
int maxCoins(vector<int>& nums) {
int n = nums.size();
vector<vector<int>> dp(n + 2, vector<int>(n + 2, 0));
vector<int> val(n + 2);
val[0] = val[n + 1] = 1;
for (int i = 1; i <= n; i++) {
val[i] = nums[i - 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++) {
int sum = val[i] * val[k] * val[j] + dp[i][k] + dp[k][j];
dp[i][j] = max(dp[i][j], sum);
}
}
}
return dp[0][n + 1];
}
};
int main()
{
Solution sol;
vector<int> nums;
int i;
while (cin >> i)
{
nums.push_back(i);
if (cin.get() == '\n') break;
}
cout << sol.maxCoins(nums) << endl;
return 0;
}