백준 27913 - SciComLove (2023) [Python] [Swift]

문제

예제 입력 1	예제 출력 1
10 1 1	2

예제 입력 2	예제 출력 2
10 3 1 2 3	2 3 4

---

코드

• Python

N, Q = map(int, input().split())
nums = set(i for i in range(N) if i % 10 in (0, 3, 6))
for _ in range(Q):
    X = int(input()) - 1
    if X in nums:
        nums.remove(X)
    else:
        nums.add(X)
    print(len(nums))

• Swift

let input = readLine()!.split(separator: " ").map { Int($0)! }
let N = input[0], Q = input[1]
var nums = Set((0..<N).filter { [0, 3, 6].contains($0 % 10) })
for _ in 0..<Q {
    let X = Int(readLine()!)! - 1
    if nums.contains(X) { nums.remove(X) }
    else { nums.insert(X) }
    print(nums.count)
}