[ Programmers / CodingTest / Python ] 점프와 순간 이동

황승환·2022년 2월 24일
0

Python

목록 보기
203/498

문제 설명

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

입출력 예

N		result
5		2
6		2
5000	5

입출력 예 설명

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

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

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

접근 방법

처음에는 다익스트라로 접근하면 풀이가 가능할 것이라고 생각하고 다익스트라로 접근하였다. 점프와 순간이동, 비용에 대한 리스트를 만들어 그때 그때의 최단 비용을 구하도록 작성하였다. 코드의 정확성은 만점을 받았지만 효율성에서 시간초과가 발생하였다.

import sys
import heapq
def solution(n):
    dj=[0, 1] # 점프 이동
    dt=[2, 1] # 순간 이동
    dc=[0, 1] # 점프 비용, 순간 이동 비용
    INF=sys.maxsize
    dist=[INF for _ in range(n+1)]
    dist[0]=0
    q=[]
    heapq.heappush(q, (0, 0))
    while q:
        cost, cur=heapq.heappop(q)
        if cost>dist[cur]:
            continue
        for i in range(2):
            nxt=(cur+dj[i])*dt[i]
            n_cost=cost+dc[i]
            if nxt<=n:
                if n_cost<dist[nxt]:
                    dist[nxt]=n_cost
                    heapq.heappush(q, (n_cost, nxt))
    ans=dist[n]
    return ans

그래서 간단한 방법으로 풀이해보기로 하였다. n을 줄여나가는 방식으로 하여 n이 짝수라면 2로 나누고, 홀수라면 1을 뺀 후, 정답 변수를 1 증가시키는 방식으로 접근하였고, 어이없게도 바로 통과되었다.

  • 정답을 저장할 변수 ans를 0으로 선언한다.
  • n이 0보다 클동안 반복하는 while문을 돌린다.
    -> 만약 n이 홀수일 경우,
    --> n을 1 감소시킨다.
    --> ans를 1 증가시킨다.
    -> 만약 n이 짝수일 경우,
    --> n을 2 나눈다.
  • ans를 반환한다.

solution.py

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

profile
꾸준함을 꿈꾸는 SW 전공 학부생의 개발 일기

0개의 댓글