一个高度为N的由正整数组成的三角形,从上走到下,求经过的数字和的最大值。
每次只能走到下一层相邻的数上,例如从第3层的6向下走,只能走到第4层的2或9上。
5
8 4
3 6 9
7 2 9 5
例子中的最优方案是:5 + 8 + 6 + 9 = 28
Input
第1行:N,N为数塔的高度。(2 <= N <= 500)
第2 - N + 1行:每行包括1层数塔的数字,第2行1个数,第3行2个数......第k+1行k个数。数与数之间用空格分隔(0 <= A[i] <= 10^5) 。
Output
输出最大值
Input示例
4
5
8 4
3 6 9
7 2 9 5
Output示例
28
#include"iostream"
#include <algorithm>
#define N 500
using namespace std;
long num[N][N], dp[N][N];
int main()
{
int pile;
cin >> pile;
for (int i = 0; i < pile; i++)
{
for (int j = 0; j <= i; j++)
{
cin >> num[i][j];
if (i == pile - 1)
{
dp[i][j] = num[i][j];
}
}
}
for (int i = pile - 2; i >= 0; i--)
{
for (int j = 0; j <= i; j++)
{
dp[i][j] = max(dp[i + 1][j], dp[i + 1][j + 1]) + num[i][j];
}
}
cout << dp[0][0] << endl;
getchar();
getchar();
return 0;
}
也可以正着做,代码主要是来自计蒜客里面的,思想都是一样的。
#include <iostream>
#include <cstdio>
using namespace std;
const int N = 100;
// 下面这个函数实现的是更新最大值,o赋值为o和x的最大值
template <class T>
void updateMax(T& o, const T& x) {
o = (o > x) ? o : x;
}
// f数组为动态规划的状态数组
// num数组为读入的数塔
// n为读入的数塔高度
int f[N][N], num[N][N], n;
int main() {
// 读入n和数塔数组num
scanf("%d", &n);
for (int i = 1; i <= n; ++i) {
for (int j = 1; j <= i; ++j) {
scanf("%d", &num[i][j]);
}
}
// step 1 begin: 在这里实现动态规划算法逻辑
for(int i = 1; i <= n; i++)
{
for(int j = 1; j <= i; j++)
{
updateMax(f[i][j], max(f[i-1][j], f[i -1][j - 1]) + num[i][j]);
}
}
// step 1 end.
// 定义最终结果变量result,因为是计算最大值,所以初始化为0
int result = 0;
for (int i = 1; i <= n; ++i) {
// step 2 begin: 在这里实现更新最终结果的逻辑
updateMax(result, f[n][i]);
// step 2 end.
}
// 输出最终最大权值和result
printf("%d\n", result);
return 0;
}