蓝桥杯python-----灌溉

题目描述

小蓝负责花园的灌溉工作。

花园可以看成一个 $n$ 行 $m$ 列的方格图形。中间有一部分位置上安装有出水管。

小蓝可以控制一个按钮同时打开所有的出水管，打开时，有出水管的位置可以被认为已经灌溉好。

每经过一分钟，水就会向四面扩展一个方格，被扩展到的方格可以被认为已经灌溉好。即如果前一分钟某一个方格被灌溉好，则下一分钟它上下左右的四个方格也被灌溉好。

给定花园水管的位置，请问 $k$ 分钟后，有多少个方格被灌溉好？

输入描述

输入的第一行包含两个整数 $n,m$。

第二行包含一个整数$t$，表示出水管的数量。

接下来 $t$ 行描述出水管的位置，其中第 $i$ 行包含两个数 $r$, $c$ 表示第 $r$ 行第 $c$ 列有一个排水管。

接下来一行包含一个整数 $k$。

其中，$1\le n,m\le 100,1\le t\le 10,1\le k\le 100$。

输出描述

输出一个整数，表示答案。

输入输出样例

输入输出样例

输入

3 6
2
2 2
3 4
1

输出

9

我的解题思想：

1. 用列表承载四个方向，便于遍历时向四周扩散
2. 每分钟遍历一次整个花园
3. 将每分钟的出水口标记为-1
4. 每分钟的被灌溉的小方格被标记为1
5. 累加有多少方格被灌溉了
6. 将标记为1的方格，重新标记为出水口
7. 进行下一分钟的灌溉操作

若有更简单的方法可以一起讨论哦！

代码如下：

# n 行 m 列
n, m = map(int, input().split())
# 创建n行m列的全为0的方格图形
l1 = [[0 for _ in range(m)] for _ in range(n)]
# t个出水口
t = int(input())
# 存储出水口的坐标，并标记为-1
a = []
for i in range(t):
    x, y = map(int, input().split())
    l1[x-1][y-1] = -1
# k分钟
k = int(input())
# 规定4个方向
directions = [
    lambda x, y: (x-1,y),
    lambda x, y: (x+1,y),
    lambda x, y: (x,y-1),
    lambda x, y: (x,y+1)
]
# 一开始就是t个方格被灌溉
s = t
for _ in range(k): # 每分钟的灌溉情况
    # 遍历方格
    for i in range(n):
        for j in range(m):
            if l1[i][j] == -1: # 若是出水口，则进行四个方向的灌溉
                for direction in range(4):
                    x, y = directions[direction](i, j)
                    if x < n and y < m and l1[x][y] == 0: # 且下个方向不能超出范围和下个方向的方格是可灌溉的
                        l1[x][y] = 1 # 下个方向标记为1
                        # print(x, y)
                        s += 1 # 累加有多少方格被灌溉了
    # 将标记为1的方格，重新标记为出水口
    for i_1 in range(n):
        for j_1 in range(m):
            if l1[i_1][j_1] == 1:
                l1[i_1][j_1] = -1
    # 进行下一分钟的灌溉操作

print(s)