Кодинг-марафон. Задача № 6.
Гарри — почтальон. У него есть почтовый участок размером n * m (матричный / 2D-список). Каждый слот в 2D-списке представляет количество писем в этом месте.
Гарри может идти только вправо и вниз. Он начинает обход в (0, 0) и заканчивает в (n-1, m-1). n представляет высоту, а m — длину матрицы.
Письма Гарри может брать только там, где находится.
Напишите функцию, возвращающую максимальное количество писем, которое Гарри может подобрать.
Примеры
harry([[5, 2], [5, 2]]) ➞ 12 # (5+5+2) harry([ [1, 2, 3, 4, 5], [6, 7, 8, 9, 10], [11, 12, 13, 14, 15] ]) ➞ 72 # (1+6+11+12+13+14+15) harry([[]]) ➞ -1
Примечание. Как вы видели в примере 3, если матрица пуста, верните -1.
Решение
def harry(po):
n, m = len(po), len(po[0])
if n == 0 or m == 0:
return -1
if n == 1 and m == 1:
return po[0][0]
for c in range(1, m):
po[0][c] += po[0][c-1]
for r in range(1, n):
po[r][0] += po[r-1][0]
for r in range(1, n):
for c in range(1, m):
po[r][c] += max(po[r-1][c], po[r][c-1])
return po[-1][-1]