(Python) - BOJ(1388번) : 바닥 장식

https://www.acmicpc.net/problem/1388

1388번: 바닥 장식

---

• 문제 : 

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

 

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

 

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

 

 

 

---

• 풀이 :

주어진 판자의 모양(가로 or 세로)에 대해 가로방향이면 가로열만, 세로방향이면 세로열만 탐색하여 같은 모양의 개수를 체크한다.

 

 

 

---

• 소스코드 : 

 

import sys
from collections import deque
input = sys.stdin.readline
 
dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]
N,M = map(int,input().split())
floor = [input() for _ in range(N)]
c = [[0]*M for _ in range(N)]
ans = 0
for a in range(N):
    for b in range(M):
        if c[a][b] == 0:
            ans += 1
            queue = deque()
            queue.append([a,b])
            c[a][b] = 1
            while queue:
                x,y = queue.popleft()
                if floor[x][y] == "-":
                    for i in range(2,4):
                        ax = x + dx[i]
                        ay = y + dy[i]
                        if 0<= ax < N and 0 <= ay < M:
                            if floor[a][b] == floor[ax][ay] and c[ax][ay]==0:
                                c[ax][ay] = 1
                                queue.append([ax,ay])
                else:
                    for i in range(2):
                        ax = x + dx[i]
                        ay = y + dy[i]
                        if 0<= ax < N and 0 <= ay < M:
                            if floor[a][b] == floor[ax][ay] and c[ax][ay]==0:
                                c[ax][ay] = 1
                                queue.append([ax,ay])
print(ans)