728x90
반응형

기본적인 BFS 문제이다.

L(땅)이 탐지될 때마다 BFS를 돌리고 카운트를 세는데 이때 C(구름)는 L이라고 생각하고 처리하면 된다.

from collections import deque

def bfs(y, x):
    q = deque()
    q.append((y, x))
    graph[y][x] = 'W'
    while q:
        y, x = q.popleft()
        for dy, dx in d:
            Y, X = y+dy, x+dx
            if (0 <= Y < r) and (0 <= X < c) and graph[Y][X] in "LC":
                q.append((Y, X))
                graph[Y][X] = 'W'

r, c = map(int, input().split())
graph = [list(input()) for _ in range(r)]
d = [(-1, 0), (1, 0), (0, -1), (0, 1)]
cnt = 0
for i in range(r):
    for j in range(c):
        if graph[i][j] == 'L':
            bfs(i, j)
            cnt += 1
print(cnt)
728x90
반응형

+ Recent posts