[백준 6359] 만취한 상범 (Swift)

6359번: 만취한 상범

중첩 반복문을 헷갈리지 않고 잘 사용하면 풀 수 있는 문제다.

• n개의 방이 있는 감옥이 주어진다
  • 처음엔 당연히 모두 잠겨 있다.
• 매 k번째 시행마다, k의 배수에 해당하는 문이 열려있다면 닫고, 닫혀 있다면 연다.
• n번째 시행까지 반복 후 열려 있는 문의 갯수를 세면 된다.
• 5 ≤ n ≤ 100 이므로 반복문으로도 충분히 풀 수 있다.

예를 들어 n=3 이라 감옥에 방이 3개일 때,
(false, false, false)
1번째 시행에는 1의 배수의 문(1, 2, 3..번째 문)이 열려있다면 닫고, 닫혀 있다면 연다. (맨 처음에는 모두 잠겨 있으므로, 모두 연다)
(true, true, true)
2번째 시행에는 2의 배수의 문이 열려있다면 닫고, 닫혀 있다면 연다.
(true, false, true)
3번째 시행에는 3의 배수의 문이 열려있다면 닫고, 닫혀 있다면 연다.
(true, false, false)
따라서 도망칠 수 있는 학생은 1명이다.

풀이

let count = Int(readLine()!)!
        (1...count).forEach { _ in
            let input = Int(readLine()!)!
            var prison = Array<Bool>(repeating: false, count: input)
            var k = 1
            while k <= input {
                (1...input).forEach { num in
                    let index = k * num - 1
                    if index < input {
                        prison[index] = !prison[index]
                    } else {
                        return
                    }
                }
                k += 1
            }
            print(prison.filter { $0 == true }.count)
        }

• prison[index] = !prison[index] 이 부분은 처음에

if prison[index] == false {
    prison[index] = true
} else {
    prison[index] = false
}

이렇게 작성했는데, 다른 풀이에서 ! 로 훨씬 짧게 표현하길래 나도 차용해 보았다. 👍