[Swift] 백준1436

import Foundation

let n = Int(readLine()!)!

/// 종말의 수 : 어떤 수에 6이 적어도 3번 이상 연속으로 들어가는 수
/// 666, 1666, 2666....9666, 10666, ..., 16660, 16661, ... , 16669
/// 구해야 하는 것 : N번째로 작은 종말의 수

var arr = [Int]()
var num = 665

while true {
    
    num += 1
    
    // 만약 666이 포함되어있으면 arr에 추가
    if num.isApocalypseNum {
        arr.append(num)
    }
    
    if arr.count >= n {
        break
    }
}

extension Int {
    
    var isApocalypseNum: Bool {
        
        var result = false
        var num = self

        while num >= 666 {
            if num % 1000 == 666 {
                result = true
                break
            }
            
            num /= 10
        }
        
        return result
    }
    
}

print(arr.last!)