Algorithm

[Swift 백준] 7785 회사에 있는 사람

Kim Roks 2025. 2. 20. 22:45

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

 

풀이 아이디어

해시 문제입니다.

최초에 출근 set, 퇴근 set을 받고, subtracting를 통해 남은 사람을 구하여 제출하였는데 오답처리가 되었습니다.

이 경우엔 중복 입장의 경우를 처리하지 못하는 게 문제였습니다.

 

예를 들어, A라는 사람이 2회 출근하고 1회 퇴근했다면 회사에 남은 사람 있어야 하는데

제 최초 풀이는 해당 과정은 처리하지 못했습니다.

 

단순히 하나의 Set에서 inset와 remove를 해주는 걸로 수정하였습니다.

 

let n = Int(readLine()!)!

var set = Set<String>()

for _ in 0..<n {
    let input = readLine()!.split(separator: " ").map{ String($0)}
    
    if input[1] == "enter" {
        set.insert(input[0])
    } else {
        set.remove(input[0])
    }
}

// 순서대로 출력을해야해서 순서가 없는 set은 그대로 사용할 수 없습니다.
var resultArr = Array(set).sorted {
    $0 > $1
}
resultArr.forEach {
    print($0)
}