[swift] 백준 - 1388번: 바닥 장식

문제 링크

1388번: 바닥 장식

---

문제

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

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

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

입력

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

출력

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

예제

---

성공 풀이

import Foundation

let NM = readLine()!.split {$0 == " "}.map {Int($0)!}, N = NM[0], M = NM[1]
let arr = (0..<N).map { _ in Array(readLine()!) }
var visited = Array(repeating: Array(repeating: false, count: M), count: N)
var result = 0

for i in 0..<N {
    for j in 0..<M {
        guard !visited[i][j] else { continue }
        var d = 1
        visited[i][j] = true
        result += 1
        if arr[i][j] == "-" {
            while j+d < M && arr[i][j+d] == "-" {
                visited[i][j+d] = true
                d += 1
            }
        } else {
            while i+d < N && arr[i+d][j] == "|" {
                visited[i+d][j] = true
                d += 1
            }
        }
    }
}
print(result)

풀이 키워드

• 바닥 장식 배열과 방문했는지 체크하는 배열을 만들었습니다.
• 바닥 장식을 모두 탐색하면서 방문하지 않은 바닥장식은 "-" 와 "|" 를 구분해서 주변에 같은 모양의 타일을 방문처리한 다음 하나의 타일로 수를 세줬습니다.