[프로그래머스] LV.2 점프와 순간 이동(python)

https://school.programmers.co.kr/learn/courses/30/lessons/12980

프로그래머스

 

문제 설명

OO 연구소는 한 번에 K 칸을 앞으로 점프하거나, (현재까지 온 거리) x 2 에 해당하는 위치로 순간이동을 할 수 있는 특수한 기능을 가진 아이언 슈트를 개발하여 판매하고 있습니다. 이 아이언 슈트는 건전지로 작동되는데, 순간이동을 하면 건전지 사용량이 줄지 않지만, 앞으로 K 칸을 점프하면 K 만큼의 건전지 사용량이 듭니다. 그러므로 아이언 슈트를 착용하고 이동할 때는 순간 이동을 하는 것이 더 효율적입니다. 아이언 슈트 구매자는 아이언 슈트를 착용하고 거리가 N 만큼 떨어져 있는 장소로 가려고 합니다. 단, 건전지 사용량을 줄이기 위해 점프로 이동하는 것은 최소로 하려고 합니다. 아이언 슈트 구매자가 이동하려는 거리 N이 주어졌을 때, 사용해야 하는 건전지 사용량의 최솟값을 return하는 solution 함수를 만들어 주세요.

예를 들어 거리가 5만큼 떨어져 있는 장소로 가려고 합니다.
아이언 슈트를 입고 거리가 5만큼 떨어져 있는 장소로 갈 수 있는 경우의 수는 여러 가지입니다.

• 처음 위치 0 에서 5 칸을 앞으로 점프하면 바로 도착하지만, 건전지 사용량이 5 만큼 듭니다.
• 처음 위치 0 에서 2 칸을 앞으로 점프한 다음 순간이동 하면 (현재까지 온 거리 : 2) x 2에 해당하는 위치로 이동할 수 있으므로 위치 4로 이동합니다. 이때 1 칸을 앞으로 점프하면 도착하므로 건전지 사용량이 3 만큼 듭니다.
• 처음 위치 0 에서 1 칸을 앞으로 점프한 다음 순간이동 하면 (현재까지 온 거리 : 1) x 2에 해당하는 위치로 이동할 수 있으므로 위치 2로 이동됩니다. 이때 다시 순간이동 하면 (현재까지 온 거리 : 2) x 2 만큼 이동할 수 있으므로 위치 4로 이동합니다. 이때 1 칸을 앞으로 점프하면 도착하므로 건전지 사용량이 2 만큼 듭니다.

위의 3가지 경우 거리가 5만큼 떨어져 있는 장소로 가기 위해서 3번째 경우가 건전지 사용량이 가장 적으므로 답은 2가 됩니다.

제한 사항

• 숫자 N: 1 이상 10억 이하의 자연수
• 숫자 K: 1 이상의 자연수

문제 해결

def solution(n):
    ans = 0
    while(n != 0):
        if n % 2 == 1:
            ans += 1
            n -= 1
        else:
            n //= 2
    return ans

 

문제 풀이 설명

처음에는 0에서부터 n까지 최적의 수를 구할려고 했었다. 하지만 애매해서 고민하던 중 n을 0까지 줄이면 해결할 수 있을까 생각해서 해결하게 됐다.

중요한 점은 순간이동을 할 때는 건전지 사용량이 카운트 되지 않는다는 점이다.

이 말은 곧 순간이동을 최대한 활용해서 문제를 해결해야 한다는 소리로 해석했고, 거리를 반으로 나누면 순간이동을 사용해 움직인 것이므로 n을 반으로 나눴다.

 

1) n을 반으로 나누었을 때 홀수인 경우:

이 때는 순간이동이 아닌 점프를 사용해 건전지 사용량을 카운트 해야한다. 다시 n이 짝수가 되게 만들면,

순간이동을 사용해 절반씩 줄이는 게 훨씬 효율적이기 때문이다!!

2) n을 반으로 나누었을 때 짝수인 경우:

이 때는 가장 효율적인 순간이동을 사용해 건전지 사용량을 카운트 하지 않고 넘어간다.

3) 결론

결국 n이 1일 때 까지 절반으로 나누게 되고, 1일 때는 점프를 사용해 n을 0으로 만들면 풀이가 끝난다.