Python算法水洼数目备战蓝桥杯

# 水洼数目
#
# 有一个大小为N×M的园子，雨后积起了水。
#
# 其中: 1代表有水, 0代表没水
#
# 八连通的积水被认为是连通在一起的。请求出园子里总共有多少水洼？（八连通指的是下图中相对w的*部分）
#
# ***
# *W*
# ***
#
# 例如某园子如图:
#
# W........WW.
# .WWW.....WWW
# ....WW...WW.
# .........WW.
# .........W..
# ..W......W..
# .W.W.....WW.
# W.W.W.....W.
# .W.W......W.
# ..W.......W.
#
# 输出3







count = 0
N = 10
M = 12
# li = []
# for i in range(10):
#     li.append(list(input()))
#
#
# print(li)
li = [['W', '.', '.', '.', '.', '.', '.', '.', '.', 'W', 'W', '.'],
      ['.', 'W', 'W', 'W', '.', '.', '.', '.', '.', 'W', 'W', 'W'],
      ['.', '.', '.', '.', 'W', 'W', '.', '.', '.', 'W', 'W', '.'],
      ['.', '.', '.', '.', '.', '.', '.', '.', '.', 'W', 'W', '.'],
      ['.', '.', '.', '.', '.', '.', '.', '.', '.', 'W', '.', '.'],
      ['.', '.', 'W', '.', '.', '.', '.', '.', '.', 'W', '.', '.'],
      ['.', 'W', '.', 'W', '.', '.', '.', '.', '.', 'W', 'W', '.'],
      ['W', '.', 'W', '.', 'W', '.', '.', '.', '.', '.', 'W', '.'],
      ['.', 'W', '.', 'W', '.', '.', '.', '.', '.', '.', 'W', '.'],
      ['.', '.', 'W', '.', '.', '.', '.', '.', '.', '.', 'W', '.']]


def dfs(li, y, x):
    li[y][x] = '.'
    for a in range(-1, 2):
        for b in range(-1, 2):
            if a == 0 and b == 0:
                continue

            if 0 <= y + a <= N-1 and 0 <= x + b <= M-1:
                if li[y + a][x + b] == 'W':
                    dfs(li, y + a, x + b)


for y in range(N):
    for x in range(M):
        if li[y][x] == 'W':
            dfs(li, y, x)
            count += 1

print(count)