长草（Python）

题目描述

小明有一块空地，他将这块空地划分为 n 行 m 列的小块，每行和每列的长度都为 1。

小明选了其中的一些小块空地，种上了草，其他小块仍然保持是空地。

这些草长得很快，每个月，草都会向外长出一些，如果一个小块种了草，则它将向自己的上、下、左、右四小块空地扩展，

这四小块空地都将变为有草的小块。请告诉小明，k 个月后空地上哪些地方有草。

输入描述

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

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

接下来包含一个整数 k。 其中，2≤n,m≤1000，1≤k≤1000。

输出描述

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

输入输出样例

示例

输入

4 5
.g...
.....
..g..
.....
2

输出

gggg.
gggg.
ggggg
.ggg.

运行限制

• 最大运行时间：1s
• 最大运行内存: 256M

代码实现 

def pd(x,y):
    if x<0 or y<0 or x>=n or y>=m:return False
    if ff[x][y]=='g':return False
    return True
def bfs():
    global k,cs,q,length
    d=[(1,0),(-1,0),(0,1),(0,-1)]
    while k>0 and length>0:
        x=yc[q][0];y=yc[q][1]
        q+=1
        length -= 1
        for i in range(4):
            dx=x+d[i][0];dy=y+d[i][1]
            if pd(dx,dy):ff[dx][dy]='g';cs+=1;yc.append((dx,dy))
        if length==0:k-=1;length=cs-q
n,m=map(int,input().split())
ff=[[0]*m for i in range(n)]
yc=[]
cs,q=0,0
for i in range(n):
    s=input()
    for j in range(m):
        ff[i][j]=s[j]
        if s[j]=='g':yc.append((i,j));cs+=1#有草的数目和坐标
length=cs-q
k=int(input())
bfs()
for i in ff:
    print(''.join(i))