[백준/파이썬/그래프/BFS] 1388번 바닥 장식 코딩테스트 연습풀이

문제 출처: https://www.acmicpc.net/problem/1388

1388번: 바닥 장식

 

문제

형택이는 건축가이다. 지금 막 형택이는 형택이의 남자 친구 기훈이의 집을 막 완성시켰다. 형택이는 기훈이 방의 바닥 장식을 디자인했고, 이제 몇 개의 나무 판자가 필요한지 궁금해졌다. 나무 판자는 크기 1의 너비를 가졌고, 양수의 길이를 가지고 있다. 기훈이 방은 직사각형 모양이고, 방 안에는 벽과 평행한 모양의 정사각형으로 나누어져 있다.

이제 ‘-’와 ‘|’로 이루어진 바닥 장식 모양이 주어진다. 만약 두 개의 ‘-’가 인접해 있고, 같은 행에 있다면, 두 개는 같은 나무 판자이고, 두 개의 ‘|’가 인접해 있고, 같은 열에 있다면, 두 개는 같은 나무 판자이다.

기훈이의 방 바닥을 장식하는데 필요한 나무 판자의 개수를 출력하는 프로그램을 작성하시오.

 

입력

첫째 줄에 방 바닥의 세로 크기N과 가로 크기 M이 주어진다. 둘째 줄부터 N개의 줄에 M개의 문자가 주어진다. 이것은 바닥 장식 모양이고, '-‘와 ’|‘로만 이루어져 있다. N과 M은 50 이하인 자연수이다.

 

출력

첫째 줄에 문제의 정답을 출력한다.

 

정답

from collections import deque

def bfs(a, b):
    deq = deque([(a,b)])
    while deq:
        x, y = deq.popleft()
        if graph[x][y] == "-":
            graph[x][y] = "1"
            if y+1 < m and graph[x][y+1] == "-":
                deq.append((x,y+1))
        elif graph[x][y] == "|":
            graph[x][y] = "1"
            if x+1 < n and graph[x+1][y] == "|":
                deq.append((x+1,y))

n, m = map(int,input().split())
cnt = 0
graph = [list(input()) for _ in range(n)]

for i in range(n):
    for j in range(m):
        if graph[i][j] != "1":
            bfs(i,j)
            cnt+=1
print(cnt)

bfs세팅을 하고 이중for문을 통해 cnt를 늘려가는 방식입니다.

cnt를 출력하면 성공