上文中使用递归来求解问题,这里我们将求解过程优化,如何优化,请继续向下看。
递归转化为递推
用一个全局数组保存结果,先将最后一行的数字放置到结果的最后一行,从倒数第二行到第一行逐层求解从i行j列元素到最后一行的路径中最大的值。
数字三角形为:
7
3 8
8 1 0
2 7 4 4
4 5 2 6 5
求解过程如下:
将数字三角形最后一行复制到全局变量最后一行,第四行第一列元素2到最后一行的路径最大值,等于2+max(4,5)=7,第四行第二列元素7到最后一行的路径最大值,等于7+max(5,2)=12,第四行第三列,4+max(2,6)=10,第四行第四列,4+max(6,4)=10,再求解第三行的每一个元素到最后一行的路径最大值。直到求解第一行第一列到最后一行的路径最大值。
#include<iostream>
#include<algorithm>
using namespace std;
#define MAX 101
int D[MAX][MAX];
int sum[MAX][MAX];
int main()
{
int n=5;
cin >> n;
for (int i = 1; i <= n; i++)
for (int j = 1; j <= i; j++)
cin >> D[i][j];
for (int j = 1; j <= n; j++)
sum[n][j] = D[n][j];
for (int i = n - 1; i >= 1; i--)
for (int j = 1; j <= i; j++)
sum[i][j] = max(sum[i+1][j],sum[i+1][j+1]) + D[i][j];
cout << sum[1][1] << endl;
return 0;
}
空间优化
没必要用二维sum数组存储每一个sum(i,j),只要从底层一行行向上递推,那么只要一维数组maxSum[100]即可,即只要存储一行的MaxSum值就可以。
数字三角形为:
7
3 8
8 1 0
2 7 4 4
4 5 2 6 5
#include<iostream>
#include<algorithm>
using namespace std;
#define MAX 101
int D[MAX][MAX];
int sum[MAX];
int main()
{
int n;
cin >> n;
for (int i = 1; i <= n; i++)
for (int j = 1; j <= i; j++)
cin >> D[i][j];
for (int i = 1; i <= n; i++)
sum[i] = D[n][i];
for (int i = n - 1; i >= 1; i--)
for (int j = 1; j <= i; j++)
sum[j] = max(sum[j],sum[j+1]) + D[i][j];
cout << sum[1] << endl;
return 0;
}
进一步考虑,连 sum 数组都可以不要,直接用 D 的第 n 行替代 sum 即可。节省空间,时间复杂度不变。
#include<iostream>
#include<algorithm>
using namespace std;
#define MAX 101
int D[MAX][MAX];
int *sum;
int main()
{
int n;
cin >> n;
for (int i = 1; i <= n; i++)
for (int j = 1; j <= i; j++)
cin >> D[i][j];
sum = D[n];//指向最后一行
for (int i = n - 1; i >= 1; i--)
for (int j = 1; j <= i; j++)
sum[j] = max(sum[j],sum[j+1]) + D[i][j];
cout << sum[1] << endl;
return 0;
}