312.戳气球

题解:https://qoogle.top/leetcode-312-burst-balloons/

. - 力扣(LeetCode)

正向思考

以 [3, 1, 5][3,1,5] 为例,对于第一次选择,我们有三种选择。

选择 3 将问题分解为左边的子问题: [][] 和 右边子问题: [1, 5][1,5]

其解等于: [] + [1, 5] + 1 * 3 * 1[]+[1,5]+1∗3∗1

选择1 将问题分解为左边的子问题: [3][3] 和 右边子问题: [5][5]

其解等于: [3] + [5] + 3 * 1 * 5[3]+[5]+3∗1∗5

选择5 将问题分解为左边的子问题: [3, 1][3,1] 和 右边子问题: [][]

其解等于: [3, 1] + [] * 1 * 5 * 1[3,1]+[]∗1∗5∗1

发现,上面的解时错的。因为正向思考的情况下,以选择 1 为例,在点爆 1 气球后,两个左右子问题并不是独立的,所以这给计算子问题带来了障碍,怎么处理才能忽略中间的 1 呢?

逆向思考

以 [3, 1, 5][3,1,5] 为例,首先拿一个气球,把这个气球当做最后一个气球,然后点爆它。这样就能够将这个气球的左右两个子问题独立开。

换种方式来说,我们选这1这个气球,然后优先点爆左边和右边的气球之后,再最后点爆这个气嘴,这时可以看出左右两个子问题是独立的,他们只和1这个气球有关联。

定义状态转移方程

dp[i][j] 从 i到 j个气球(闭区间)能够获取的最大的硬币数量

dp[i][j] = dp[i][k - 1] + dp[k + 1][j] + nums[i - 1] * nums[k] * nums[j + 1] (i <= k <= j)

这种方式仍然是O(n!),但是由于利用了“记忆化”的特性,将重复的子问题写在了内存中,保证每一种子问题只计算一遍。这时能够通过,但是还可以优化。

class Solution {
private:
    vector<vector<int>> dp;
    int helper(vector<int>& nums, int i, int j) {
        // boundary
        if (i > j) return 0;
        if (dp[i][j] > 0) return dp[i][j];
        
        // search
        for (int k = i; k <= j; ++k) {
            int left = helper(nums, i, k - 1);
            int right = helper(nums, k + 1, j);
            int delta = nums[k] * nums[i - 1] * nums[j + 1];
            dp[i][j] = max(dp[i][j], left + right + delta);
        }
        return dp[i][j];
    }
public:
    int maxCoins(vector<int>& nums) {
        int n = nums.size();
        nums.insert(nums.begin(), 1);
        nums.push_back(1);
        dp = vector<vector<int>>(n + 2, vector<int>(n + 2, 0));
        int ans = helper(nums, 1, n);
        return ans;
    }
};

class Solution {
public:
    /**
     * @param nums: A list of integer
     * @return: An integer, maximum coins
     */
    int maxCoins(vector<int> &nums) {
        // write your code here
        if (nums.size() <= 0) {
            return 0;
        }
        int size = nums.size()+2;
        nums.insert(nums.begin(), 1);
        nums.push_back(1);
        // i,j之间的吹爆的和是多少,不包含i,j位置
        std::vector<std::vector<int>> dp(size, std::vector<int>(size, 0));
        for (int len = 3; len <= nums.size(); ++len) {
            for (int i = 0; i + len - 1 < nums.size(); ++i) {
                int j = i + len - 1;
                for (int k = i+1; k < j; ++k) {
                    dp[i][j] = max(dp[i][j], dp[i][k] + dp[k][j] + nums[i] * nums[k] * nums[j]);
                }
            }
        }
        return dp[0][size-1];
    }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值