Python DP算法 龙与地下城游戏问题

问题描述:

给定一个二维数组map，含义是一张地图，例如如下，矩阵：
　　-2 　-3　 3
　　-5　-10　1
　　0 　30　-5
　　
游戏规则如下：
　　骑士从左上角出发，每次只能向右或者向下走，最后到达右下角见到公主。
地图中每个位置的只代表骑士要遭遇的事。如果是负数，表示此处有怪兽，要让骑士损失血量。如果是非负数，表示此处有血瓶，能让骑士回血。
骑士从左上角到右下角的过程，走到任何一个位置，血量都不能少于１。
为了保证骑士能顺利见到公主，初始血量至少是多少？根据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)