[LeetCode]312. 戳气球

最新推荐文章于 2025-06-27 21:52:16 发布

原创最新推荐文章于 2025-06-27 21:52:16 发布 · 382 阅读

0 ·

CC 4.0 BY-SA版权

DP 专栏收录该内容

10 篇文章

订阅专栏

本文介绍了一种使用动态规划解决312.戳气球问题的方法。通过构建二维数组f来存储子问题的解，利用状态转移方程f[i][j]=max{f[i][k-1]+f[k+1][j]+nums[i-1]*nums[k]*nums[j+1]}

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

DP
f[i][j]=max { f[i][k-1] + f[k+1][j] + nums[i-1] * nums[k] * nums[j+1] }
戳破 i~j 的气球，最后一个戳破的气球是 k

312. 戳气球

class Solution {
public:
    int maxCoins(vector<int>& nums) {
        int n=nums.size();
        if (n==0) return 0;
        //初始化
        int f[n+5][n+5];
        for (int i=0;i<n;i++)
            for (int j=0;j<n;j++) f[i][j]=0;
        for (int i=0;i<n;i++){
            if (i==0 && i==n-1) return nums[0];
            if (i==0) f[i][i]=nums[i+1]*nums[i];
            else if (i==n-1) f[i][i]=nums[i-1]*nums[i];
            else f[i][i]=nums[i-1]*nums[i]*nums[i+1];
        }
        
        for (int p=2;p<=n;p++)
            for (int i=0;i+p-1<n;i++){
                int j=i+p-1;
                //k表示最后一次被扎破的气球
                for (int k=i;k<=j;k++){
                    int help=0;
                    if (i==0){
                        if (j==n-1) help=1;
                        else help=nums[j+1];
                    }
                    else if (j==n-1){
                        help=nums[i-1];
                    }
                    else help=nums[i-1]*nums[j+1];
                    if (k==i){
                        f[i][j]=max(f[k+1][j]+nums[k]*help,f[i][j]);
                        continue;
                    }
                    if (k==j){
                        f[i][j]=max(f[i][k-1]+nums[k]*help,f[i][j]);
                        continue;
                    }
                    f[i][j]=max(f[i][k-1]+f[k+1][j]+help*nums[k],f[i][j]);
                }
            }
        return f[0][n-1];
    }
};