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
- 因为题目要求为:有多条路径,靠右的路径优先(即仅仅输出靠右的路径即可,无需多条路径都输出)。
- 所以此题若存在多条路径,如果按常规思路进行从上到下进行动态规划时,易出现遍历到最后一行时,某一列为多条路径的最后终点,而无法求出题目的第二要求:输出该路径上的数字。
- 所以我们可以换种方式:逆向求状态转移方程,即从下往上。好处是如果从上往下查看求完的 dp 数组,不用到最后一行才知道路径值总和最大是哪个,而是从一开头就可以知道了,因为从上往下沿着 dp 值大的那条路径往下遍历即可,因为这必定是路径总和最大的路径。
3. 算法解题思路
- 初始化输入数组
- 初始化 dp 数组,由于是逆向递推,所以初始化的是最后一行数据
- 求解每一次状态,具体值是下一行中左边或右边大的那个,加上当前索引的值即可,即 dp[i][j] = max(dp[i + 1][j], dp[i + 1][j + 1]) + a[i][j];
- 求完 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前端成长记录