lintcode 168. 吹气球【记忆化搜索+区间dp】

168. 吹气球

中文English

有n个气球，编号为0到n-1，每个气球都有一个分数，存在nums数组中。每次吹气球i可以得到的分数为 nums[left] * nums[i] * nums[right]，left和right分别表示i气球相邻的两个气球。当i气球被吹爆后，其左右两气球即为相邻。要求吹爆所有气球，得到最多的分数。

样例

样例 1:

输入：[4, 1, 5, 10]
输出：270
解释：
nums = [4, 1, 5, 10] 吹爆 1, 得分 4 * 1 * 5 = 20
nums = [4, 5, 10]   吹爆 5, 得分 4 * 5 * 10 = 200 
nums = [4, 10]   吹爆 4, 得分 1 * 4 * 10 = 40
nums = [10]   吹爆 10, 得分 1 * 10 * 1 = 10
总得分 20 + 200 + 40 + 10 = 270

样例 2:

输入：[3,1,5]
输出：35
解释：
nums = [3, 1, 5] 吹爆 1, 得分 3 * 1 * 5 = 15
nums = [3, 5] 吹爆 3, 得分 1 * 3 * 5 = 15
nums = [5] 吹爆 5, 得分 1 * 5 * 1 = 5
总得分 15 + 15 + 5  = 35

注意事项

1. 你可以假设nums[-1] = nums[n] = 1。-1和n位置上的气球不真实存在，因此不能吹爆它们。
2. 0 ≤ n ≤ 500, 0 ≤ nums[i] ≤ 100

这个题告诉我们

欠的技术宅早晚要还

区间dp基础题 放在面试里面都算是难的了

for循环里面是遍历某一个点，这个点是最后打的气球

所以dp[x][y]=dp[x][i-1] + dp[i+1][y] + left*right*nums[i]

注意的是 flag要在for循环后面！因为for结束了，才是当前区间遍历完

class Solution {
public:
    /**
     * @param nums: A list of integer
     * @return: An integer, maximum coins
     */
    int dfs (int x, int y, vector<int> &nums, vector<vector<int> > &dp, vector<vector<bool> > &flag) {
        if (flag[x][y]) {
            return dp[x][y];
        }
        int res = 0;
        for (int i = x; i <= y; i ++) {
            dfs(x, i-1,nums,dp,flag);
            dfs(i+1, y,nums,dp,flag);
            int left = nums[x-1], right = nums[y+1];
            res = max(res, dp[x][i-1] + dp[i+1][y] + left*right*nums[i]);
        }
        dp[x][y] = res;
        flag[x][y] = true;
        return dp[x][y];
    }
    int maxCoins(vector<int> &nums) {
        // write your code here
        int n = nums.size();
        if (n < 1) {
            return 0;
        }
        nums.insert(nums.begin(), 1);
        nums.push_back(1);
        vector<vector<int>>dp(n+2,vector<int>(n+2,1));
        vector<vector<bool>>flag(n+2,vector<bool>(n+2,false));
        return dfs(1,n,nums,dp,flag);
    }
};