一道简单的DFS题
题目描述
上图给出了一个数字三角形。从三角形的顶部到底部有很多条不同的路径。对于每条路径,把路径上面的数加起来可以得到一个和,你的任务就是找到最大的和。
路径上的每一步只能从一个数走到下一层和它最近的左边的那个数或者右 边的那个数。此外,向左下走的次数与向右下走的次数相差不能超过 1。
输入描述
输入的第一行包含一个整数 (1≤N≤100),表示三角形的行数。
下面的 N 行给出数字三角形。数字三角形上的数都是 0 至 100 之间的整数。
输出描述
输出一个整数,表示答案。
输入输出样例
输入
5
7
3 8
8 1 0
2 7 4 4
4 5 2 6 5
输出
27
思路
直接深度搜索找到所有可能的情况下的最大和
用0填充数组是计算更加方便
代码
import os
import sys
# 请在此输入您的代码
n=int(input())
dfs=[[0 for i in range(2*n-1)] for j in range(n)]
for i in range(n):
number=list(map(int,input().split()))
start=4-i
for j in range(i+1):
dfs[i][start]=number[j]
start+=2
Max=0
def con(m,n,l,r,nums):
global dfs,Max
nums+=dfs[m][n]
if m==len(dfs)-1:
if l-r==1 or l-r==-1 or l-r==0:
Max=max(Max,nums)
else:
con(m+1,n-1,l+1,r,nums)
con(m+1,n+1,l,r+1,nums)
con(0,n-1,0,0,0)
print(Max)
编码不易,有帮助的话点个赞支持一下吧~~~