问题描述:
给定一个二维数组map,含义是一张地图,例如如下,矩阵:
-2 -3 3
-5 -10 1
0 30 -5
游戏规则如下:
骑士从左上角出发,每次只能向右或者向下走,最后到达右下角见到公主。
地图中每个位置的只代表骑士要遭遇的事。如果是负数,表示此处有怪兽,要让骑士损失血量。如果是非负数,表示此处有血瓶,能让骑士回血。
骑士从左上角到右下角的过程,走到任何一个位置,血量都不能少于1。
为了保证骑士能顺利见到公主,初始血量至少是多少?根据map,返回初始血量。
解法:
# -*- coding: utf-8 -*-
# [-2(Knight Position), -3, 3]
# [-5, -10, 1]
# [0, 30, -5(Destination)]
#
# HP >= 1
# DP Matrix: Record History
#
# [0, 0, 0]
# [0, 0, 0]
# [0, 0, 6]
def calculate_min_HP(map_matrix):
row = len(map_matrix)
column = len(map_matrix[0])
# init DP matrix
DP_matrix = [[0 for i in range(column)] for j in range(row)]
DP_matrix[row-1][column-1] = max(-map_matrix[row-1][column-1]+1, 1)
for row_index in range(row-2, -1, -1):
DP_matrix[row_index][column-1] = max(DP_matrix[row_index+1][column-1] - map_matrix[row_index][column-1], 1)
for column_index in range(column-2, -1, -1):
DP_matrix[row-1][column_index] = max(DP_matrix[row-1][column_index+1] - map_matrix[row-1][column_index], 1)
for row_index in range(row-2, -1, -1):
for column_index in range(column-2, -1, -1):
right_value = max(DP_matrix[row_index][column_index+1] - map_matrix[row_index][column_index], 1)
down_value = max(DP_matrix[row_index+1][column_index] - map_matrix[row_index][column_index], 1)
DP_matrix[row_index][column_index] = min(right_value, down_value)
print ("----------------------")
for row in DP_matrix:
print row
print DP_matrix[0][0]
return DP_matrix
if __name__ == '__main__':
map_matrix = [[-2, -3, 3], [-5, -10, 1], [0, 30, -5]]
calculate_min_HP(map_matrix)