leetcode 312 戳气球（区间dp）

最新推荐文章于 2024-12-10 20:13:14 发布

原创最新推荐文章于 2024-12-10 20:13:14 发布 · 302 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

33 篇文章

订阅专栏

探讨一种通过动态规划解决复杂问题的方法：如何在戳破一系列气球时，通过特定策略最大化收集硬币的数量。文章详细解释了区间动态规划的应用，通过递归函数实现最优解的计算。

312. 戳气球

难度困难297

有 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

首先我们可以想到区间DP，假设dp[l][r]为区间[l,r]产生的最大硬币数量，接下来考虑转移，区间dp转移时候一般都是考虑在哪里切一刀，然后划分为子区间，这里我们切一刀的意思转化为(l,r)区间内剩余的一个气球，从而打到转移的目的。为了我们计算的方便，我们的[l,r]转为(l,r)来讨论。

dp[l][r] = max(dp[l][i] + dp[i][r] + arr[l]*arr[i]*arr[r]) for i in len(arr)

class Solution {
public:
    vector<vector<int>> dp;
    vector<int> arrmv;
    int dfs(int l,int r){
        if(r<=l+1)return 0;
        if(dp[l][r]!=-1)return dp[l][r];
        dp[l][r] = 0;
        for(int i = l+1;i<r;i++)
            dp[l][r] = max(dp[l][r], arrmv[i]*arrmv[l]*arrmv[r] + dfs(l,i)+dfs(i,r));
        return dp[l][r];
    }
    int maxCoins(vector<int>& nums) {  
        int  n  = nums.size();
        dp.assign(n+2,vector<int>(n+2,-1));
        nums.insert(nums.begin(),1);
        nums.insert(nums.end(),1);
        arrmv=nums;
        return dfs(0,n+1);
    }
};