[프로그래머스/C++] 점프와 순간 이동

미숙한 블로그 주인이 코딩테스트 문제를 풀어가는 과정을 담은 글입니다. 이 풀이가 효율적인 풀이가 아닐 수 있으며, 부정확한 정보가 많이 있을 수 있습니다. 보완해야할 점이 있다면 댓글로 남겨주세요!

 

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

 

풀이

입출력 예시

 

풀이

규칙만 찾았다면 어렵지 않은 문제였다. 0에서 부터 시작하는 것이 아닌, 주어진 수에서 시작해서 규칙을 찾아가면 되었다.

 

5를 예로 들면

5 = 4 + 1

4 = 2 * 2

2 = 2 * 1

1 = 0 + 1 // 1회 점프

 

5000을 예로 들면

5000 = 2 * 2500

2500 = 2 * 1250

1250 = 2 * 625

625 = 624 + 1 // 1회 점프

.

.

.

4 = 2 * 2

2 = 2 * 1

1 = 0 + 1 // 5회 점프

 

int solution(int n) {
    int ans = 0;
    
    while (n > 0) {
        if (n % 2 == 0) {
            n /= 2;
        }
        else {
            n--;
            ans++;
        }
    }

    return ans;
}

 

결과

 

 

+ 또 다른 규칙

채점이 끝나고 다른 사람의 코드를 보던 중, 신비한 방법을 알아내었다. 시프트 연산으로 비트 1의 수를 세아리는 방법이었는데, 놀랍게도 비트 1의 수와 답이 일치했다.

 

ex)

5000은 이진수로 1001110001000 이다. 1의 수가 5개이다.

6은 이진수로 110 이다. 1의 수가 2개이다.

 

그래서 아래와 같이 매우 짧은 코드로도 풀 수 있었다.

 

#include <bitset>

using namespace std;

int solution(int n) {
    bitset<31> b(n);
        
    return b.count();
}