题目1-python华为原题题库-亲子游戏

华为 OD 机试：亲子游戏

题目描述：

宝宝和妈妈参加亲子游戏，在一个二维矩阵（N ∗ N）的格子地图上，宝宝和妈妈抽签决定各自的位置，地图上每个格子有不同的糖果数量，部分格子有障碍物。

游戏规则是妈妈必须在最短的时间（每个单位时间只能走一步）到达宝宝的位置，路上的所有糖果都可以拿走，不能走障碍物的格子，只能上下左右走。

请问妈妈在最短到达宝宝位置的时间内最多拿到多少糖果（优先考虑最短时间到达的情况下尽可能多拿糖果）。

输入描述

第一行输入为 N，N 标识二维矩阵的大小
之后 N 行，每行有 N 个值，表格矩阵每个位置的值。

其中：

• -3：妈妈
• -2：宝宝
• -1：障碍
• >=0：糖果数(0 表示没有糖果，但是可以走)

输出描述

输出妈妈在最短到达宝宝位置的时间内最多拿到多少糖果，行末无多余空格。

备注

地图最大 50*50

示例一

输入

4
3 2 1 -3
1 -1 1 1
1 1 -1 2
-2 1 2 3

输出

9

说明

此地图有两条最短路径可到宝宝位置，绿色线和黄色线都是最短路径 6 步，但黄色拿到的糖果更多，9 个

示例二

输入

4
3 2 1 -3
-1 -1 1 1
1 1 -1 2
-2 1 -1 3

输出

-1

说明

此地图妈妈无法到达宝宝位置

解题思路

最短路径很容易使用 BFS 计算得到。本题难点在于如何计算最短路径下的最多糖果。

题解

import queue

num = 4
# 地图填充
dt = """
3 2 1 -3
1 -1 14 5
1 1 1 2
-2 1 2 3
""".strip()
dt = [list(map(int, i.split())) for i in dt.split('\n')]
# 走法队列,格式 x坐标,y坐标,糖果数量cc
zf = queue.Queue()
# 检查是否走过
check = [[0] * num for _ in range(num)]
for i in range(num):
    for j in range(num):
        # 母亲坐标填充队列
        if dt[i][j] == -3:
            mx, my = i, j
            zf.put((mx, my, 0))
            check[mx][my] = 1
db = 0
jg = []
while not zf.empty():
    zfc = zf.qsize()
    tl = []
    for _ in range(zfc):
        x, y, cc = zf.get()
        for zx, zy in [[0, 1], [0, -1], [1, 0], [-1, 0]]:
            tx, ty = x + zx, y + zy
            # print(tx, ty)
            if 0 <= tx < num and 0 <= ty < num and dt[tx][ty] != -1 and check[tx][ty] == 0:
                # print(tx, ty)
                if dt[tx][ty] == -2:
                    jg.append([db, cc])
                    break
                check[tx][ty] = 1
                tl.append((tx, ty, cc + dt[tx][ty]))
    for i in (sorted(tl, key=lambda x: -x[2])):
        zf.put(i)
    db += 1
if not jg:
    print(-1)
else:
    zb = min([i[0] for i in jg])
    zc = max([i[1] for i in jg if i[0] == zb])
    print(zc)