蓝桥杯 经典 bfs 灌溉 python

题目描述

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

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

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

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

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

输入描述

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

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

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

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

其中

1≤n,m≤100,1≤t≤10,1≤k≤100。

输出描述

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

输入输出样例

示例 1

输入

3 6
2
2 2
3 4
1

输出

9

 

​
from queue import Queue#用到队列
n, m = map(int, input().split())
t = int(input())
la = []
for i in range(t):
    a, b = map(int, input().split())
    la.append((a, b))
k = int(input())
f = [[0] * m for i in range(n)]#n行m列的放个图形
def right(x, y):#判断是否超过边界
    if x >= 0 and x < n and y >= 0 and y < m:
        return True
    else:
        return False
vis = [[0] * m for i in range(n)]#是否访问过
dir = [[-1, 0], [1, 0], [0, 1], [0, -1]]#四个方向
sum = 0
def bfs():
    global sum  #在函数内用全局变量，sum 记录被灌溉数
    q = Queue()
    for l in la:
        x = l[0] - 1
        y = l[1] - 1
        q.put((x, y, 0))
        vis[x][y] = 1
        sum = sum + 1

    while q.empty() == False:
        now = q.get()#now是个三元组，now[0]为x,now[1]为y,now[2]为第几分钟
        if now[2] == k:
            return
        for i in range(4):
            x1 = now[0] + dir[i][0]
            y1 = now[1] + dir[i][1]
            if right(x1, y1) and vis[x1][y1] == 0:
                q.put((x1, y1, now[2] + 1))
                vis[x1][y1] = 1
                sum = sum + 1
bfs()
print(sum)

​