算法设计与分析 SCAU10303 数字三角（优先做）

10303 数字三角（优先做）

时间限制:1000MS 代码长度限制:10KB
提交次数:0 通过次数:0

题型: 编程题 语言: G++;GCC;VC;JAVA

Description

问题描述：给定一个由n行数字组成的数字三角形，如下图所示。试用动态规划算法，计算出从三角
顶部至底部的一条路径，使得该路径经过的数字总和最大。

注意每个数字只能走向下一行左边或右边的数字，而不能跳跃的走。

     7
  3   8
8   1   0

2 7 4 4
4 5 2 6 5

输入格式

第一行是数字三角的行数n，1<=n<=100。接下来n行是数字三角各行中的数字，所有数字在0~99之间。

输出格式

输出两行，第一行是计算出的最大路径的和值，第二行是该路径上的数字。若有多条路径，靠右的路径
优先（即仅仅输出靠右的路径即可，无需多条路径都输出）。

如：
Input:
5
7
3 8
8 1 6
2 7 4 4
4 5 2 4 5

有两条路径：7-3-8-7-5和7-8-6-4-5都为30，由于后者靠右，因此仅输出后者。

Output:
30
7 8 6 4 5

输入样例

5
7
3 8
8 1 0
2 7 4 4
4 5 2 6 5

输出样例

30
7 3 8 7 5

解题思路

动态规划

题目要求为最大路径的和值以及路径，我们很容易得出解出此题的基本思路：动态规划，每次更新 dp 数组时，对于向下一行左边或右边的数字，选择较大的那个即可保证路径是最大。

但由于此题有特殊情况：若有多条路径，靠右的路径
优先（即仅仅输出靠右的路径即可，无需多条路径都输出）。因此需要稍微转换一下 dp 的思路

1. dp 方程定义

dp[i][j] 表示路径计算至第 i 行第 j 列元素时，此时的最大路径值。

2. 状态转移方程

从上到下，从左到右进行循环遍历，每次遍历时，在下一行左边或右边的数字中选择较大的那个即可，因此状态转移方程如下：

dp[i][j] = max(dp[i + 1][j], dp[i + 1][j + 1]) + a[i][j]; // i < n

1. 因为题目要求为：有多条路径，靠右的路径优先（即仅仅输出靠右的路径即可，无需多条路径都输出）。
2. 所以此题若存在多条路径，如果按常规思路进行从上到下进行动态规划时，易出现遍历到最后一行时，某一列为多条路径的最后终点，而无法求出题目的第二要求：输出该路径上的数字。
3. 所以我们可以换种方式：逆向求状态转移方程，即从下往上。好处是如果从上往下查看求完的 dp 数组，不用到最后一行才知道路径值总和最大是哪个，而是从一开头就可以知道了，因为从上往下沿着 dp 值大的那条路径往下遍历即可，因为这必定是路径总和最大的路径。

3. 算法解题思路

1. 初始化输入数组
2. 初始化 dp 数组，由于是逆向递推，所以初始化的是最后一行数据
3. 求解每一次状态，具体值是下一行中左边或右边大的那个，加上当前索引的值即可，即 dp[i][j] = max(dp[i + 1][j], dp[i + 1][j + 1]) + a[i][j];
4. 求完 dp 数组后，再从上往下、从左到右遍历，沿着 dp 值大的那条路径往下遍历即可，i 控制遍历到的行数，j 控制遍历到的列数，一边遍历一边打印即可

更多注释可查看下方的完整代码中，有助于理解

代码如下

#include <iostream>

using namespace std;

int a[101][101];
int dp[101][101];


int main()
{
    int i, j, n;
    cin >> n;

    // 按下面输入的三角形，下一行左边或右边的数字在坐标上表现为：对于 a[i][j]，下一行左边或右边为 a[i + 1][j] 和 a[i + 1][j + 1]
    for(i = 1; i <= n; i++) {
        for(j = 1; j <= i; j++) {
            cin >> a[i][j];
        }
    }

    // 由于是从下往上，即反过来动态规划，所以要先将最底层进行初始化
    for(i = 1; i <= n; i++) {
        dp[n][i] = a[n][i];
    }

    for(i = n - 1; i >= 1; i--) {
        for(j = 1; j <= i; j++) {
            // 选大的那个作为路径
            dp[i][j] = max(dp[i + 1][j], dp[i + 1][j + 1]) + a[i][j];
        }
    }

    cout << dp[1][1] << endl; // dp 数组最顶层元素即为路径最大值

    // 开始找最大值的路径，即从上往下，从左到右找动态规划值更大的那个，一定就是路径，因为是从那里往上冒的
    cout << a[1][1] << " ";
    j = 1;

    for(i = 2; i <= n; i++) {
        // 找dp值大的那个，如果一样大，由于题目要求输入最右端路径，所以输出右边那个
        if(dp[i][j] <= dp[i][j + 1]) {
            cout << a[i][++j] << " ";
        } else {
            cout << a[i][j] << " ";
        }
    }


    return 0;
}

最后

对我感兴趣的小伙伴可查看以下链接

• 我的掘金主页：https://juejin.cn/user/1302297507801358
• 博客主页：http://blog.zhangjiancong.top/
• 公众号：Smooth前端成长记录