灌溉（Python）

题目描述

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

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

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

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

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

输入描述

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

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

接下来 t 行描述出水管的位置，其中第 i 行包含两个数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

运行限制

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

def pd(dx,dy):
  if 0<=dx<n and 0<=dy<m and vis[dx][dy]==0:return True
  return False

def bfs():
   global c,s,length,t,vis,k
   d=[(1,0),(-1,0),(0,-1),(0,1)]
   while k>0 and length>0:
     x,y=s[c][0],s[c][1]
     c+=1
     for i in range(4):
       dx,dy=d[i][0]+x,d[i][1]+y
       if pd(dx,dy):t+=1;vis[dx][dy]=1;s.append((dx,dy))
     length-=1
     if length==0:k-=1;length=t-c


n,m=map(int,input().split())#n行m列
t=int(input())#被灌溉的土地数目
s=[];vis=[[0]*m for i in range(n)];c=0
for i in range(t):
  ri,ci=map(int,input().split())#水管坐标，可理解为一开始就被灌溉的草地
  s.append((ri-1,ci-1))
  vis[ri-1][ci-1]=1#对被灌溉的土地做标记
k=int(input())#灌溉时间
length=t-c
bfs()
print(t)