蓝桥杯真题之BFS——种草

题目描述

小明有一块空地，他将这块空地划分为 n 行 m 列的小块，每行和每列的长度都为 1。小明选了其中的一些小块空地，种上了草，其他小块仍然保持是空地。这些草长得很快，每个月，草都会向外长出一些，如果一个小块种了草，则它将向自己的上、下、左、右四小块空地扩展，这四小块空地都将变为有草的小块。请告诉小明，k 个月后空地上哪些地方有草。

输入描述

输入的第一行包含两个整数 n, m。接下来 n 行，每行包含 mm 个字母，表示初始的空地状态，字母之间没有空格。如果为小数点，表示为空地，如果字母为 g，表示种了草。接下来包含一个整数 k

输出描述

输出 n 行，每行包含 m 个字母，表示 k 个月后空地的状态。如果为小数点，表示为空地，如果字母为 g，表示长了草。

 思路：定义一个列表，比如mat[],用它来记录草地的实时情况，定义一个队列，用来扩展新的草地。每出一个队列的数据，就把新的长出草来的列表，队列内容更新。

from collections import deque

n,m = map(int,input().split()) #输入n,m
mat = [] #定义列表，用来存储初始草地的状态
q = deque() #定义队列，用来记录长草的位置坐标

for i in range(n): #依次输入，初始化mat
    mat.append(list(input()))
k = int(input()) #k个月
#接下来，依次遍历所有坐标，如果哪块草地有草，就把对应位置的坐标入队
for i in range(n):
    for j in range(m):
        if mat[i][j]=='g':
            q.append((i,j))
#定义四个方向（上下左右
directions = [(0,1),(0,-1),(1,0),(-1,0)]
#定义宽度优先函数
def bfs():
    t = len(q)
    while t>0:
        temp = q.popleft() #出队
        for d in directions:
            nx = temp[0]+d[0] #得到新的行坐标
            ny = temp[1]+d[1] #得到新的列坐标
            if nx>=0 and nx<n and ny>=0 and ny<m and mat[nx][ny]!='q': 
                mat[nx][ny]='q' #更新mat内的内容
                q.append((nx,ny)) #更新队列
        t-=1

for i in range(k):
    bfs()
for i in range(n):
    print(''.join(mat[i]))