[프로그래머스][Python] 점프와 순간이동

최더디·2021년 2월 14일
0
post-thumbnail

📃 문제 설명

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 이상의 자연수

입출력 예

Nresult
52
62
50005

입출력 예 설명
입출력 예 #1
위의 예시와 같습니다.

입출력 예 #2
처음 위치 0 에서 1 칸을 앞으로 점프한 다음 순간이동 하면 (현재까지 온 거리 : 1) x 2에 해당하는 위치로 이동할 수 있으므로 위치 2로 이동합니다. 이때 1 칸을 앞으로 점프하면 위치3으로 이동합니다. 이때 다시 순간이동 하면 (현재까지 온 거리 : 3) x 2 이동할 수 있으므로 위치 6에 도착합니다. 이 경우가 건전지 사용량이 가장 적으므로 2를 반환해주면 됩니다.

입출력 예 #3
위와 같은 방식으로 합니다.



💻 문제 풀이

def solution(n):
    power = 1           # 무조건 1칸은 가야하므로 건전지 사용량 1로 초기화
    while True:
        if n == 1:      
            break
        # 2로 나눴을 때 나누어 떨어지면 순간이동이 가능함.
        power += n%2    # n%2 에서 나머지 값을 계속해서 사용량에 더해준다.
        n = n//2        # n은 계속해서 n//2 를 통해 몫을 넣어준다.
    
    return power

중요 포인트

  • 순간 이동을 최대한 활용해야 하는 작업이다.(순간이동은 x2를 하는 작업)
    그렇다면 반대로 도착지(=n)에서 "나누기 2"를 통해 최대한 순간이동을 통해 거꾸로 간다고 생각했다.
  • 이때 2를 나눠주면서 나머지 값 (0 or 1)은 계속해서 더해준다.
  • 나머지 값이 0이라면 순간이동을 통해 갈 수 있는 위치.
    나머지 값이 1이라면 power를 써야하는 위치.
  • 계속해서 n//2를 해줌으로써 n==1이 되었을 때 반복문을 빠져나오면 된다.

다른 사람 풀이

결국 이 문제는 2진수로 바꾼 후 1의 갯수를 카운팅하면 더 쉽게 해결할 수 있는 문제였다.
내가 풀었던 방법을 조금 더 생각해보면 2진수 계산이라는 것을 알 수 있었다..

def solution(n):
    return bin(n).count('1')    # 2진수로 바꾼 후 1의 갯수를 카운팅
profile
focus on why

0개의 댓글