分治算法-气球游戏问题

这学期上算法设计与分析课，遇到一道分治算法下的气球游戏问题。

题目描述

　　刚刚今天去游乐场玩，发现了一个新的游戏项目，游戏是这样的，场上一共有 n 个气球，它们的编号是0到n-1，然后每个气球上还有一个数字，我们使用数组nums来保存这些数字。

　　现在游戏要求刚刚戳破所有的气球。每当刚刚戳破一个气球i时，刚刚可以获得nums[left] * nums[i] * nums[right]个积分。这里的left和right指的是和i相邻的两个气球的序号。（注意每当刚刚戳破了气球i后，气球left和气球right就变成了相邻的气球。）

　　求所能获得积分的最大值。

输入数据

　　输入中有若干组测试样例，第一行为一个正整数T(T≤1000)，表示测试样例组数。每组测试样例包含2部分： 第一部分有一行，包含1个正整数n（0≤n≤500），第二部分为一行，有n个数，第i个数表示num[i],(0≤num[i]≤100)。

输出数据

对每组测试数据，单独输出一行答案，表示最大的积分值。

样例输入

1

4

3 1 5 8

样例输出

167

题意理解

题意很好读懂，戳破所有气球，每戳破一个气球i获得nums[left] * nums[i] * nums[right]个积分。求怎么戳得分最多。样例里，先戳气球1，得分3*5=15；再戳气球5，得分3*5*8=120；再戳气球3，得分3*8=24；最后戳气球8，得分8。加起来就是15+120+24+8=167

代码

#include <stdio.h>
#include <math.h>

// 定义最大气球数量
#define MAX_N 500

int maxCoins(int nums[], int n) {
    int i, j, k;
    int maxCoins = 0;

    // 创建一个二维数组dp，用于保存中间结果
    int dp[MAX_N][MAX_N];

    // 初始化dp数组为0
    for (i = 0; i < n; i++) {
        for (j = 0; j < n; j++) {
            dp[i][j] = 0;
        }
    }

    // 从长度为1的子问题开始，逐渐扩展到长度为n的问题
    for (int len = 1; len <= n; len++) {
        for (i = 0; i <= n - len; i++) {
            j = i + len - 1;
            for (k = i; k <= j; k++) {
                int leftValue = (i > 0) ? nums[i - 1] : 1;
                int rightValue = (j < n - 1) ? nums[j + 1] : 1;
                int before = (i <= k - 1) ? dp[i][k - 1] : 0;
                int after = (k + 1 <= j) ? dp[k + 1][j] : 0;
                dp[i][j] = fmax(dp[i][j], before + (leftValue * nums[k] * rightValue) + after);
            }
        }
    }

    return dp[0][n - 1];
}

int main() {
    int T;
    scanf("%d", &T);

    while (T--) {
        int n;
        scanf("%d", &n);
        int nums[MAX_N];

        for (int i = 0; i < n; i++) {
            scanf("%d", &nums[i]);
        }

        int result = maxCoins(nums, n);
        printf("%d\n", result);
    }

    return 0;
}

代码分析

这段代码是精髓：

for (int len = 1; len <= n; len++) {
        for (i = 0; i <= n - len; i++) {
            j = i + len - 1;
            for (k = i; k <= j; k++) {
                int leftValue = (i > 0) ? nums[i - 1] : 1;
                int rightValue = (j < n - 1) ? nums[j + 1] : 1;
                int before = (i <= k - 1) ? dp[i][k - 1] : 0;
                int after = (k + 1 <= j) ? dp[k + 1][j] : 0;
                dp[i][j] = fmax(dp[i][j], before + (leftValue * nums[k] * rightValue) + after);
            }
        }
    }

从长度为1扩展至长度为n的问题。

leftValue用来存左侧的数值，rightValue用来存右侧的数值；before用来存上次操作的数值，after用来存下次操作的数值。最后dp[i][j]维护最大值。

感悟

该题来源：北京交通大学《算法设计与分析》课程第二次作业-分治 A题。虽然放在分治的作业下，但颇有动态规划的味道。