【问题描述】
给定一个数塔,如下图所示。在此数塔中,从顶部出发,在每一节点可以选择走左下或右下,一直走到底层。请找出一条路径,使路径上的数值和最大。
9 | ||||||||
12 | 15 | |||||||
10 | 6 | 8 | ||||||
2 | 18 | 9 | 5 | |||||
19 | 7 | 10 | 4 | 16 |
【输入形式】
输入时第一行一个整数n,表示该数塔的行数,其余n行表示该塔每行的数值
【输出形式】
输出包含两行,第一行为最大路径上的数值之和, 第二行n个数字为从上而下最大路径数值
【样例输入】
5 9 12 15 10 6 8 2 18 9 5 19 7 10 4 16
【样例输出】
59 9 12 10 18 10
【算法思想】
在这个数塔问题中,最小子问题是指塔的最底层,也就是包含单个元素的行。
在这个问题中,每个单独的节点(最底层的元素)可以看作一个子问题。最底层的节点不需要做选择,它们本身就是路径的终点,因此它们的最大路径数值就是它们自身的值。
最底层只有一个节点,最大路径数值就是9,因为只有一个节点,路径的和就等于该节点的值。
然后,通过向上计算,我们可以将当前行的每个节点转化为下一行的子问题。在计算过程中,每个节点都将选择下一行中相邻的两个节点中较大的那个,并加上自己的值,这样我们逐步向上计算出每个节点的最大路径数值。这样的递推过程直到达到塔顶,得到了从塔顶到底层的最大路径数值。
所以,最小子问题在这里指的是数塔的最底层,它是动态规划递推过程的起点。通过解决这个最小子问题,我们逐步扩展到更大的子问题,最终得到整个数塔的最优解。
创建一个二维数组dp
,大小为n x n,用来存储从每个节点出发到底层的最大路径数值。
从塔的最底层开始,将每个节点的值存储在dp
数组的最后一行,即dp[n-1][i] = tower[n-1][i]
。
从倒数第二行开始向上遍历,对每个节点计算从该节点出发到底层的最大路径数值。对于每个节点dp[i][j]
,可以选择向左下或者右下走,所以从当前节点出发到底层的最大路径数值为dp[i][j] = tower[i][j] + max(dp[i+1][j], dp[i+1][j+1])
,其中tower[i][j]
表示当前节点的值。
最后,dp[0][0]
就是从塔顶开始出发到底层的最大路径数值。我们还可以根据dp
数组逆向推导出最大路径上的节点值。
#include <vector>
#include <iostream>
using namespace std;
int main() {
int n;
cin >> n;
vector<vector<int>> tra; // 存储输入的数塔
vector<vector<int>> dp; // 存储最大路径数值
for (int i = 1; i <= n; i++) {
vector<int> x(i);
for (int j = 0; j < i; j++) {
cin >> x[j];
}
tra.push_back(x);
dp.push_back(x);
}
// 从底层向上计算每个节点的最大路径数值
for (int i = n - 2; i >= 0; i--) {
for (int j = 0; j < i + 1; j++) {
dp[i][j] += max(dp[i + 1][j], dp[i + 1][j + 1]);
}
}
cout << dp[0][0] << endl; // 输出从塔顶出发到底层的最大路径数值
int row = 0, col = 0;
cout << tra[row][col] << " "; // 输出塔顶节点的值
for (int i = 0; i < n - 1; i++) {
// 根据dp数组的值选择走左下还是右下
if (dp[row + 1][col] > dp[row + 1][col + 1]) {
row++;
} else {
row++;
col++;
}
cout << tra[row][col] << " "; // 输出路径上的节点值
}
return 0;
}