[백준 18312] 시각 (Swift)

이코테(이것이 코딩 테스트다) 구현 예제 중 하나. (113페이지)

18312번: 시각

• 0-n시, 0-59분, 0-59초 사이의 모든 시간을 살펴봐야 하므로 3중 반복문으로 구한다.
• 주의할 점은, 한 자리 숫자일 때 앞에 0이 생기는 것까지 고려해야 한다는 것이다. 그래야 K가 0인 경우를 놓치지 않고 셀 수 있다.
  • 나는 appendZero라는 별도 함수로 분리했다.

풀이1(69104KB, 56ms)

let input = readLine()!.split(separator: " ")
let n = Int(input[0])!
let k = Character(String(input[1]))
var answer = 0

func appendZero(to num: Int) -> String {
    var answer = ""
    if num < 10 {
        answer = "0\(num)"
    } else {
        answer = "\(num)"
    }
    return answer
}

(0...n).forEach { hour in
    (0...59).forEach { minute in
        (0...59).forEach { second in
            let time = "\(appendZero(to: hour))\(appendZero(to: minute))\(appendZero(to: second))"
            if time.contains(k) {
                answer += 1
            }
        }
    }
}
print(answer)

풀이2(69104KB, 12ms)

다른 풀이를 보니 시, 분, 초 모두 2자리임을 이용해 십의 자리(= 10으로 나눈 값)과 일의 자리(10으로 나눈 나머지)를 일일이 비교해 구했다.

이쪽은 문자열을 사용하지 않고 순수 연산으로 구할 수 있어서 실행시간이 훨씬 빨랐다.

let input = readLine()!.split(separator: " ").map{ Int($0)! }
let n = input[0]
let k = input[1]
var answer = 0
(0...n).forEach { hour in
    (0...59).forEach { minute in
        (0...59).forEach { second in
            if hour / 10 == k || hour % 10 == k
                || minute / 10 == k || minute % 10 == k
                || second / 10 == k || second % 10 == k {
                answer += 1
            }
        }
    }
}
print(answer)