题目地址:灌溉
题目整体不是很难,跟深度广度优先遍历差不多写法,将被蔓延的方格当做水管,但要注意在第一次遍历时,可能将从上方蔓延下来的方格当做水管,而这个蔓延下来的方格是下一次遍历时我们所要当做的水管,所以我们要用到备用数组储存这一次循环所蔓延的方格,最后再将原数组与备用数组置同。
完整代码:
n, m = map(int, input().split())
waterpipe = int(input()) # 水管个数
ground = [[0] * m for _ in range(n)] # 方格图形
_ground = [[0] * m for _ in range(n)] # 备用数组,防止将此次蔓延灌溉误认为是水管
for i in range(waterpipe): # 添加水管
x, y = map(int, input().split())
ground[x - 1][y - 1] = 1
time = int(input())
ans = 0
def check(x, y):
if ground[x][y] == 1: # 说明是水管
_ground[x][y] = 1
if (x - 1) >= 0: # 向左蔓延
_ground[x - 1][y] = 1
if (x + 1) < n: # 向右蔓延
_ground[x + 1][y] = 1
if (y + 1) < m: # 向下蔓延
_ground[x][y + 1] = 1
if (y - 1) >= 0: # 向上蔓延
_ground[x][y - 1] = 1
gournd = _ground # 置同
for i in range(time):
for x in range(n):
for y in range(m):
check(x, y)
for grounds in _ground:
for i in grounds:
if i == 1:
ans += 1
print(ans)