백준 - 랭킹전 대기열(Swift)

문제 이름

 

 

 문제 설명

 

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

20006번: 랭킹전 대기열

 

 나의 풀이

 

단순한 구현문제였지만 시간이 오래걸렸다.

난 구현문제에 많이 약한 것 같은 느낌이 든다.

 

이 문제를 푸는데 시간이 오래걸렸던 요인은 두 가지이다.

 

1. 문제 이해 -> 문제를 꼼꼼히 읽고 입력값과 출력값을 예상해야하는데 입력값과 출력값만 보고 문제를 이해하려고했다. 문제자체에서 입력값과 출력값을 이해하게 하려고 노력하니까 문제를 잘 읽자

2. 자료구조 이용 -> 처음에는 rooms변수를 [방번호: [플레이어]]형식으로 자료구조를 하려고 했는데 방이 들어온 순서대로를 구현하려고 할 때 어려움이 있어 Room이라는 구조체를 따로 만들었다. 처음 한 방식은 딕셔너리이기 때문에 들어온 순서대로 정렬이 되어있지 않다. 그렇기에 Room구조체를 빠르게 생성해야한다는 게 포인트였던 것 같다.

 

import Foundation

let input = readLine()!.components(separatedBy:" ").map{Int($0)!}
let N = input[0], M = input[1] // 플레이어의 수, 방의 제한

struct Player {
    let level: Int
    let name: String
}

struct Room {
    let roomRange: Int // 방의 입장가능 레벨범위
    var roomList: [Player] //방에 들어간 플레이어들
    let roomId: Int // 방 생성순으로
}

// 방의 상태
var rooms = [Room]()

var idx = 0 // 방의 순서
for i in 0..<N {
    let input = readLine()!.components(separatedBy:" ")
    let p1 = Player(level: Int(input[0])!, name: input[1])
    // 레벨범위에 맞고 방의 수용치가 꽉차지 않고 생성된 날짜가 가장 빠른순으로
    let tempRoom = rooms.filter{ $0.roomRange-10 <= p1.level && $0.roomRange+10 >= p1.level }
                        .filter{ $0.roomList.count < M }
                        .sorted(by: { $0.roomId < $1.roomId })
                           
    if tempRoom.isEmpty {
        idx += 1
        rooms.append(Room(roomRange: p1.level, roomList: [p1], roomId: idx))
    } else { // 방이 비어있지 않다면
        let foundIdx = rooms.firstIndex(where: {$0.roomId == tempRoom[0].roomId})!
        rooms[foundIdx].roomList.append(p1)
    }
}

for x in rooms {
    if x.roomList.count == M {
        print("Started!")
    } else {
        print("Waiting!")
    }
    let temp = x.roomList.sorted(by: {$0.name < $1.name })
    temp.forEach { print("\($0.level) \($0.name)")}
}