蓝桥杯 | 数字三角形 问题 | 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
运行限制
最大运行时间:1s
最大运行内存: 256M
- 问题结构分析
为了方便处理,我们将上图数字三角形改成直角三角形,如下:
i\j | 0 | 1 | 2 | 3 | 4 |
---|---|---|---|---|---|
0 | 7 | ||||
1 | 3 | 8 | |||
2 | 8 | 1 | 0 | ||
3 | 2 | 7 | 4 | 4 | |
4 | 4 | 5 | 2 | 6 | 5 |
这样向左下走就变为了向下走,向右下走还是向右下走。
用 a[ i ][ j ] 来表示第 i 行第 j 列的数字,i 是行索引,j 是列索引。
定义 dp[ i ][ j ] 表示从 (0,0) 走到 (i,j) 的最大和。
-
递推关系建立(见代码)
-
最优方案追踪
① n-1为偶数时:dp[n-1] [ (n-1)//2 ]
向下走了 n - 1 = 5 - 1 = 4 层
两种方式 | 向下走 | 向右下走 | 相差不能超过 1 |
---|---|---|---|
j=0时 | 4个 | 0个 | ✘ |
j=1时 | 3个 | 1个 | ✘ |
j=2时 | 2个 | 2个 | ✔ |
j=3时 | 1个 | 3个 | ✘ |
j=4时 | 0个 | 4个 | ✘ |
② n-1为奇数时:max( dp[n-1] [ (n-1)//2 ], dp[n-1] [ (n-1)//2+1 ] )
向下走了 n - 1 = 4 - 1 = 3 层
两种方式 | 向下走 | 向右下走 | 相差不能超过 1 |
---|---|---|---|
j=0时 | 3个 | 0个 | ✘ |
j=1时 | 2个 | 1个 | ✔ |
j=2时 | 1个 | 2个 | ✔ |
j=3时 | 0个 | 3个 | ✘ |
- 代码
import os
import sys
# 请在此输入您的代码
dp=[]
n=int(input()) # 输入行数n
for i in range(n): # 初始化
dp.append(list(map(int,input().split())))
'''
7
3 8
8 1 0
2 7 4 4
4 5 2 6 5
>>> print(dp)
[[7], [3, 8], [8, 1, 0], [2, 7, 4, 4], [4, 5, 2, 6, 5]]
'''
for i in range(n):
for j in range(i+1):
# 递推关系建立
if i==0: # 在第0行,dp[i][j] = dp[0][0]
dp[i][j] = dp[i][j]
elif j==0: # 在第0列,最左边元素只能由正上方得到
dp[i][j] += dp[i-1][j]
elif j==i: # 在对角线列,最右边元素只能由左上方得
dp[i][j] += dp[i-1][j-1]
else: # 最一般情况,其余元素由上方两边较大值得到
dp[i][j] += max(dp[i-1][j],dp[i-1][j-1])
# 满足约束条件:向左下走的次数与向右下走的次数相差不能超过 1
if (n-1)%2==0: # n-1为偶数时
print(dp[n-1][(n-1)//2])
else: # n-1为奇数时
print(max(dp[n-1][(n-1)//2],dp[n-1][(n-1)//2+1])) # (n-1)//2为向上取整,(n-1)//2+1为向下取整