BOJ-2810 컵홀더 Swift

문제

십년이면 강산이 변한다.

강산이네 동네에 드디어 극장이 생겼고, 강산이는 극장에 놀러갔다. 매점에서 콜라를 산 뒤, 자리에 앉은 강산이는 큰 혼란에 빠졌다. 양쪽 컵홀더를 이미 옆 사람들이 차지했기 때문에 콜라를 꽂을 컵 홀더가 없었기 때문이다. 영화를 보는 내내 콜라를 손에 들고 있던 강산이는 극장에 다시 왔을 때는 꼭 콜라를 컵 홀더에 놓겠다는 다짐을 한 후 집에 돌아갔다.

극장의 한 줄에는 자리가 N개가 있다. 서로 인접한 좌석 사이에는 컵홀더가 하나씩 있고, 양 끝 좌석에는 컵홀더가 하나씩 더 있다. 또, 이 극장에는 커플석이 있다. 커플석 사이에는 컵홀더가 없다.

극장의 한 줄의 정보가 주어진다. 이때, 이 줄에 사람들이 모두 앉았을 때, 컵홀더에 컵을 꽂을 수 있는 최대 사람의 수를 구하는 프로그램을 작성하시오. 모든 사람은 컵을 한 개만 들고 있고, 자신의 좌석의 양 옆에 있는 컵홀더에만 컵을 꽂을 수 있다.

S는 일반 좌석, L은 커플석을 의미하며, L은 항상 두개씩 쌍으로 주어진다.

어떤 좌석의 배치가 SLLLLSSLL일때, 컵홀더를 *로 표시하면 아래와 같다.

*S*LL*LL*S*S*LL*

위의 예에서 적어도 두 명은 컵홀더를 사용할 수 없다.

입력

첫째 줄에 좌석의 수 N이 주어진다. (1 ≤ N ≤ 50) 둘째 줄에는 좌석의 정보가 주어진다.

출력

컵을 컵홀더에 놓을 수 있는 최대 사람의 수를 출력한다.

내가 푼 풀이

- 주어진 문자열을 일반석이면 주변에 양쪽에 *하나를, 커플석이면 두 개의 쌍 양쪽으로 *을 넣어준다

- 좌석과 상관없이 양 끝 좌석에 컵홀더 하나씩 있으므로, 오른쪽이든 왼쪽이든 한쪽을 기준으로 컵홀더를 모두 써가면 최대로 이용할 수 있다.

- 컵홀더까지 포함된 배열에서 두개씩 추출하여 좌석과 컵홀더를 포함하면 카운트해 주고, 커플석처럼 컵홀더를 포함하지 않으면 한자리를 넘겨서 세준다.

 

해당 조건을 토대로 코드로 구현한다.

 

import Foundation

let count = Int(readLine()!)!
var sit = Array(readLine()!).map{ String($0)}
var set = ["*"]
var idx = 0
var result = 0

// 주어진 문자열을 컵홀더위치까지 넣어준다.
while idx < sit.count {
    if sit[idx] == "S" {
        set.append(sit[idx])
        set.append("*")
        idx += 1
    } else {
        set.append(sit[idx])
        set.append(sit[idx+1])
        set.append("*")
        idx += 2
    }
}
idx = 0

// 컵홀더까지 포함된 문자열에서 두개씩 쌍으로 추출하여 좌석과 컵홀더로 존재하면 카운트
// 커플석과같이 L L 이 추출되면 한칸 건너뛰어서 재추출한다.
while idx < set.count - 1 {
    var s1 = set[idx]
    var s2 = set[idx+1]
    if s1 == "*" || s2 == "*" {
        result += 1
        idx += 2
    } else {
        idx += 1
    }
}

print(result)