백준 - 기차가 어둠을 헤치고 은하수를(Swift)

백준 - 기차가 어둠을 헤치고 은하수를(Swift)

 

 

 문제 설명

 

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

15787번: 기차가 어둠을 헤치고 은하수를

 

 나의 풀이

 

입력으로 100,000이 들어오지만 기차에서 좌석을 한 칸씩 당기거나 미는 정도의 행위만 하기 때문에 O(n)으로 처리할 수 있는 문제다.

그렇기 때문에 요구사항대로 구현만 잘 해주면 문제는 쉽게 풀린다.

 

import Foundation

struct Train: Hashable {
    var seats: [Int]
}

var trains = [Train]()
var availThrough = Set<Train>()
let nm = readLine()!.components(separatedBy:" ").map{Int($0)!}
let N = nm[0], M = nm[1]

for _ in 0..<N {
    trains.append(Train(seats: Array(repeating:0, count: 20)))
}

for _ in 0..<M {
    let input = readLine()!.components(separatedBy:" ").map{Int($0)!}
    let type = input[0]
    
    switch type {
    case 1: // i번째 기차에 x번 좌석에 태움
        trains[input[1]-1].seats[input[2]-1] = 1
    case 2: // i번째 기차에 x번 좌석 비움
        trains[input[1]-1].seats[input[2]-1] = 0
    case 3: // 모든 좌석 한 칸씩 뒤로
        trains[input[1]-1].seats.removeLast()
        trains[input[1]-1].seats.insert(0, at:0)
    case 4: // 모든 좌석 한 칸씩 앞으로
        trains[input[1]-1].seats.removeFirst()
        trains[input[1]-1].seats.append(0)
    default: break
    }    
}

for train in trains {
    availThrough.insert(train)
}
print(availThrough.count)