[蓝桥杯2020初赛] 数字三角形Python版

题目描述

上图给出了一个数字三角形。从三角形的顶部到底部有很多条不同的路径。
对于每条路径，把路径上面的数加起来可以得到一个和，你的任务就是找到最大的和。
路径上的每一步只能从一个数走到下一层和它最近的左边的那个数或者右边的那个数。
此外，向左下走的次数与向右下走的次数相差不能超过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

#[蓝桥杯2020初赛] 数字三角形Python版
N = int(input())
matrix = []
for i in range(N):
    matrix.append(list(map(int, input().split())))
for i in range(1,N):
    for j in range(i+1):
        if j == 0:
            matrix[i][j] += matrix[i-1][j]
        elif j == i:
            matrix[i][j] += matrix[i-1][j-1]
        else:
            matrix[i][j] += max(matrix[i - 1][j - 1], matrix[i - 1][j])
print(matrix[N - 1][N // 2] if N % 2 == 1 else max(matrix[N - 1][N // 2 - 1], matrix[N - 1][N // 2]))