【蓝桥杯 每天一道算法题】数字三角形

题目描述

上图给出了一个数字三角形。从三角形的顶部到底部有很多条不同的路径。对于每条路径，把路径上面的数加起来可以得到一个和，你的任务就是找到最大的和。

路径上的每一步只能从一个数走到下一层和它最近的左边的那个数或者右 边的那个数。此外，向左下走的次数与向右下走的次数相差不能超过 1。

输入描述

输入的第一行包含一个整数 N (1≤N≤100)，表示三角形的行数。

下面的 N 行给出数字三角形。数字三角形上的数都是 0 至 100 之间的整数。

输出描述

输出一个整数，表示答案。

输入输出样例

示例

输入

5
7
3 8
8 1 0
2 7 4 4
4 5 2 6 5

输出

27

运行限制

• 最大运行时间：1s
• 最大运行内存: 256M

答案

# https://www.lanqiao.cn/problems/505/learning/
N = int(input())
martrix = [[] for _ in range(N)]
for i in range(N):
    martrix[i] = list(map(int, input().split()))
dp = martrix.copy()
for i in range(1, N):
    for j in range(i + 1):
        if j == 0:
            dp[i][j] = dp[i - 1][j] + martrix[i][j]
        elif j == i:
            dp[i][j] = dp[i - 1][j - 1] + martrix[i][j]
        else:
            dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - 1]) + martrix[i][j]
if N % 2 == 1:
    print(dp[-1][N // 2])
else:
    print(max(dp[-1][N // 2], dp[-1][N // 2 - 1]))