算法-动态规划1-数塔问题
题意:
下图是一个数塔,从顶部出发在每一个节点可以选择向左或者向右走,一直走到底层,要求找出一条路径,使得路径上的数字之和最大.
算法实现:
I. 首先利用一个二维数组data存储数塔的原始数据,然后利用一个中间数组dp存储每一次决策过程中的结果。
II. 初始化dp,将data的最后一层拷贝到dp中。dp[n][j] = data[n][j] (j = 1, 2, …, n) 其中,n为数塔的层数。
III. 在动态规划过程汇总,我们有dp[i][j] = max(dp[i+1][j], dp[i+1][j+1]) + data[i][j],最后的结果保存在dp[0][0]中。
代码:
int DataTower(int d[n][n]) {
int maxadd[n][n] = { 0 }, path[n][n] = { 0 }; //初始化
//maxadd记录和,path记录每层中j的选择
int i, j;
for (j = 0; j < n; j++) { //初始化底层决策
maxadd[n-1][j] = d[n-1][j];
}
for (i = n-2; i >= 0; i++) {
for (j = 0; j <= i; j++) {
if (maxadd[i + 1][j] > maxadd[i + 1][j + 1]) {
maxadd[i][j] = d[i][j] + maxadd[i + 1][j]; //上面的元素本身+下面两个元素中比较小的那个
path[i][j] = j; //本次决策选择下标为j的元素
}
else {
maxadd[i][j] = d[i][j] + maxadd[i + 1][j + 1];
path[i][j] = j + 1; //选择下标为j+1的元素
}
}
}
cout << "路径为:" << d[0][0]; //输出最顶层的元素
j = path[0][0];
for (i = 1; i < n; i++) {
cout << "-->" << d[i][j];
j = path[i][j];
}
return maxadd[0][0]; //返回数塔的最大数值和,以及其路径
}