题目描述
小明有一块空地,他将这块空地划分为 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]))