1388번: 바닥 장식 (8 / 28) - 실버 4

문제

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

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

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

입력

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

출력

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

예제 입력 1

4 4
----
----
----
----

예제 출력 1

4

예제 입력 2

6 9
-||--||--
--||--||-
|--||--||
||--||--|
-||--||--
--||--||-

예제 출력 2

31

예제 입력 3

7 8
--------
|------|
||----||
|||--|||
||----||
|------|
--------

예제 출력 3

13

예제 입력 4

10 10
||-||-|||-
||--||||||
-|-|||||||
-|-||-||-|
||--|-||||
||||||-||-
|-||||||||
||||||||||
||---|--||
-||-||||||

예제 출력 4

41

예제 입력 5

6 6
-||--|
||||||
|||-|-
-||||-
||||-|
||-||-

예제 출력 5

19

코드

import sys

# n, m 인풋 받기
n, m = map(int, sys.stdin.readline().split())
# 인접 리스트 생성
graph = [list(sys.stdin.readline().strip()) for _ in range(n)]  # 개행을 지우려면 strip() 사용
# 2차원 방문 여부 리스트 초기화
visited = [[False for _ in range(m)] for _ in range(n)]  # 이거 숙지 하기
count = 0
def count_board(row, column, direction):
    visited[row][column] = True

    # 만약 방향이 '-' 이고 column + 1이 m보다 작고 방문을 하지 않았으며 '-' 값을 가진다면 다음 컬럼 인덱스로 재귀함수를 호출한다.
    if direction == '-' and column + 1 < m and not visited[row][column + 1] and graph[row][column + 1] == '-':
        count_board(row, column + 1, direction)
    # 만약 방향이 '|' 이고 row + 1이 n보다 작고 방문을 하지 않았으며 '|' 값을 가진다면 다음 행 인덱스로 재귀함수를 호출한다.
    if direction == '|' and row + 1 < n and not visited[row + 1][column] and graph[row + 1][column] == '|':
        count_board(row + 1, column, direction)

for i in range(n):
    for j in range(m):
        if not visited[i][j]:
            count += 1
            if graph[i][j] == '-':
                count_board(i, j, '-')
            elif graph[i][j] == '|':
                count_board(i, j, '|')

print(count)

코드 풀이

1. [list(sys.stdin.readline().strip()) for _ in range(n)]
   • n만큼 반복을 수행하며 2차원 list를 만든다.
   • strip을 사용한 이유는 sys.stdin.readline() 개행을 제거하기 위함
2. [[False for _ in range(m)] for _ in range(n)]
   • 방문 여부를 확인하기 위해 그래프와 동일한 크기의 2차원 배열을 초기화함
   • [[False] * n for _ in range(n)] 도 많이 사용됨
3. count_board(row, column, direction) 재귀 함수를 초기화
   • visited[row][column] = True 방문했다고 표시
   • 방향성 체크
     • -
   • 다음 인덱스의 바운더리 체크
     • column + 1 < m
   • 다음 인덱스의 방문 여부 체크
     • not visited[row][column + 1]
   • 다음 인덱스의 방향 체크
     • graph[row][column + 1]
4. 2중 for문을 사용하여 모든 배열을 탐색하여 -과 |의 개수를 센다.

핵심 포인트

1. DFS 문제에서 방문여부를 체크하는 것은 매우 중요하다. 반드시 초기화 할 것
2. 해당 문제의 경우, 모든 2차원 배열을 탐색하여 -과 |의 개수를 세어야 하는 문제이다. 따라서 2중 for 루프를 사용하여 모든 위치를 방문한다.