[백준/Swift] 1436번: 영화감독 숌

 

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

1436번: 영화감독 숌

 

이 문제는 예전에 c++로 풀었던 문제인데 완전 탐색 공부하면서 다시 풀어본다.

i를 666부터 차례로 올라가면서 666을 연속으로 가지고 있는지 체크하는 방식으로 했다.

 

그런데 가끔씩 이런 문제는 분명 맞았는데... 시간초과가 나오는 경우가 있다.

최대 10000번째 종말의 수를 구해도 충분히 시간 내에 구할 수 있을 텐데 왜 시간초과가??

 

 

[시간 초과]

import Foundation

let n = Int(readLine()!)!
var m = 0
var i = 666
while true {
    let tmp = String(i)
    if tmp.contains("666") { m += 1 } //666이 포함되어있나?
    if n == m { break }
    i += 1
}
print(i)

 

이렇게 문자열 내에 "666"가 있는지 확인하는 contains 함수를 사용했던 게 시간을 잡아먹었는지

 

import Foundation

let n = Int(readLine()!)!
var m = 0
var i = 666
while true {
    let str = String(i)
    var cnt = 0
    for s in str {
        if s == "6" { cnt += 1 }
        else { cnt = 0 }
        if cnt == 3 {
            m += 1
        }
    }
    if n == m { break }
    i += 1
}
print(i)

 

이렇게 하니까 해결됐다! 꼭 이렇게 풀어서 구현해야.. 되는 Swift... 이 쪽으로 좀 더 공부를 해봐야겠다.

빠르게 진행하기 위해!!