백준 A->B(Swift)

백준 A->B(Swift)

 

 

 문제 설명

 

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

16953번: A → B

 

 나의 풀이

 

처음에 n의 마지막 숫자가 1로끝나는 경우와 2로 나눠떨어지는 경우 두 가지만 체크해서 정답이 안 나왔었는데 두개의 경우가 아닌 경우도 있기 때문에 그 경우를 else { break }로 체크해줬더니 문제가 해결되었다.

 

import Foundation

let input = readLine()!.components(separatedBy: " ").map{Int($0)!}
var target = input[0], n = input[1]
var cnt = 1
while n > target {
    cnt += 1
    if String(n).last! == "1" {
        n = Int(String(n).prefix(String(n).count-1))!
    } else {
        if n % 2 == 0 {
            n = n / 2    
        } else {
            break
        }
    }
}
if n == target { print(cnt) }
else { print(-1) }

 

 다른 사람의 풀이

 

1의자리를 10으로 나눠서 구한 풀이다. 스위프트에서 위의 내 풀이처럼 문자열로 다루려고하면 다소 복잡한 부분이 있기 때문에 숫자에서 처리할 수 있으면 되도록 숫자에서 처리하는 게 좋다.

 

import Foundation

let input = readLine()!.components(separatedBy: " ").map{Int($0)!}
var target = input[0], n = input[1]
var cnt = 1
while n > target {
    if n%10 == 1 {
        n = n/10
    } else if n%2 == 0 {
        n = n/2
    } else {
        break
    }
    cnt += 1
}
if n == target { print(cnt) }
else { print(-1) }

 

 피드백