蓝桥云课——灌溉 Python

 

---

题目地址：灌溉

        题目整体不是很难，跟深度广度优先遍历差不多写法，将被蔓延的方格当做水管，但要注意在第一次遍历时，可能将从上方蔓延下来的方格当做水管，而这个蔓延下来的方格是下一次遍历时我们所要当做的水管，所以我们要用到备用数组储存这一次循环所蔓延的方格，最后再将原数组与备用数组置同。

完整代码：

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)