戳气球

问题描述 :
有 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 ≤ 300, 0 ≤ nums[i] ≤ 100

示例:
输入: [3,1,5,8]
输出: 167
解释: nums = [3,1,5,8] --> [3,5,8] --> [3,8] --> [8] --> []
coins = 315 + 358 + 138 + 181 = 167

输入说明 :
首先输入气球数(即nums数组长度) n
然后输入n个整数，表示气球上的数字nums[i]
0 ≤ n ≤ 300, 0 ≤ nums[i] ≤ 100

输出说明 :
输出一个整数

输入范例 :
4
3 1 5 8

输出范例 :
167

#include<iostream>
#include<vector>

using namespace std;
int maxCoins(vector<int> &nums){
	int n = nums.size();
        vector<int> tmp(n + 2);
        tmp[0] = tmp[n + 1] = 1;
        for (int i = 1; i <= n; i ++) tmp[i] = nums[i - 1];
        vector<vector<int>> dp(n + 2, vector<int>(n + 2, 0));
        for (int step = 1; step <= n; step ++) {
            for (int i = 1; i + step - 1 <= n; i ++) {
                int j = i + step - 1;
                for (int k = i; k <= j; k ++) {
                    dp[i][j] = max(dp[i][j], dp[i][k - 1] + dp[k + 1][j] + tmp[i - 1] * tmp[k] * tmp[j + 1]);
                }
            }
        }
        return dp[1][n];
} 
int main(){
	int n;
	cin>>n;
	vector<int> nums;
	int tmp;
	for(int i=0;i<n;i++){
		cin>>tmp;
		nums.push_back(tmp);
	}
	int res=maxCoins(nums);
	cout<<res;
}