본문 바로가기

알고리즘/프로그래머스

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

문제내용

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

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

문제풀이

가장 처음 생각했던 방법은 dfs를 이용해 모든 경우의수를 찾는것.

그래서 n까지 1씩 증가시키고 카운트를 해줬고, n까지 왔다면 다시 되돌아 가면서 순간이동(곱하기)을 하면서 위치가 n이 될때 최솟값을 갱신하도록 로직을 짰다.

그리고 결과적으로 틀렸다. 출력값은 맞지만 너무 많은 경우의 수를 생각해야하고, 제한사항에 n은 최대 10억까지 가능하기 때문에 dfs로 모든 경우의수를 세는건 무리였다.

 

그리고 다른사람 풀이를 보고 생각한건데 생각보다 경우의수를 줄일수 있는 방법은 매우 간단하다.

어차피 최종 위치는 n에 도달해야한다는 점을 이용하면 n을 시작으로해서

n이 짝수면 나누기 2.

n이 홀수면 -1을 해줌으로써 n이 딱 0이 될때까지 반복 계산을 해주면된다.

 

이렇게 하면 좋은게 0부터 n까지의 경우의 수를 세려고하면 n을 넘어갔을때의 경우들도 고려해야하기 때문에 n을 시작으로해서 줄이는게 좋다.

 


코드

틀린거(완탐)

public class Solution {
	static private int min;
    public int solution(int n) {
        min = Integer.MAX_VALUE;
        dfs(n,1,1);
        return min;
    }
    
    public static void dfs(int n, int use, int loca) {
    	if(n==loca) {
    		min = Math.min(min, use);
    		return;
    	}
    	
	dfs(n, use+1, loca+1);
    	if(loca*2<=n) dfs(n, use, loca*2);
    }
}

 

맞는거(나누기,빼기)

public class Solution {
	static private int min;
    public int solution(int n) {
        dfs(n,0);
        System.out.println(min);
        return min;
    }
    public static void dfs(int n,int use) {
    	if(n==0) {
    		min = use;
    		return;
    	}
    	
    	if(n%2==0) dfs(n/2,use);
    	else if(n%2!=0) dfs(n-1,use+1);
    	return;
    }
}

마치며

생각만해낸다면 쉽게 풀수 있는문제다. 이문제를 풀고 다시 느낀건 어떤식으로 문제에 접근해야할지 신중하게 고민해볼 필요가 있다는걸 다시한번 느꼈다..