[알고리즘 PS] 백준 5014번 스타트링크 자바 문제풀이

문제

해당 포스팅은 백준의 5014번 스타트 링크 의 접근과 해결 방법을 설명한 글 입니다.
정답 소스 코드를 확인하시려면 solve url 에서 확인 가능합니다.

5014번: 스타트링크

 

이 문제를 해결하기 위해 어떤 방식으로 접근해야 하는지를 먼저 생각해보자.

 

문제 접근

 

이번 문제는 각각 층 수로 갈 수 있는 최단 거리를 출력하면 되는 문제이다.

 

해결법

 

현재 층에서 목표로 하는 층 수가 같다면 0을 출력하고 엘레베이터로 갈 수 있는 수는 현재 층에서 +u 해주고 -d 를 해준다.

 

그리고 각각의 단계에서 depth 를 설정하고 다음 depth 는 이전 depth + 1 이여야 한다.

 

만약 초기 depth 를 1으로 하려 한다면 마지막 결과 출력할 때 -1 을 잊지 않도록 하자.

 

그리고 엘레베이터로 갈 수 없는 층수와 엘레베이터를 탈 필요가 없는 층은 분명한 차이가 있다.

 

정답 코드

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        String[] fsgud = br.readLine().split(" ");

        int f = Integer.parseInt(fsgud[0]); // 총 층수
        int s = Integer.parseInt(fsgud[1]); // 시작 층
        int g = Integer.parseInt(fsgud[2]); // 목표 층
        int u = Integer.parseInt(fsgud[3]); // 한 번에 올라갈 수 있는 수
        int d = Integer.parseInt(fsgud[4]); // 한 번에 내려갈 수 있는 수

        int[] upAndDown = {u, -1 * d};

        int[] visited = new int[f + 1];

        Queue<Integer> queue = new LinkedList<>();

        queue.add(s);
        visited[s] = 1;

        while(!queue.isEmpty()) {
            int front = queue.remove();

            for (int i = 0; i < 2; i++) {

                int next = front + upAndDown[i];

                if(0 < next && next <= f) {
                    if(visited[next] == 0) {
                        visited[next] = visited[front] + 1;
                        queue.add(next);
                    }
                }
            }
        }

        if (visited[g] != 0) {
            bw.write(String.valueOf(visited[g] - 1));
        }
        else if(s == g) {
            bw.write("0");
        } else {
            bw.write("use the stairs");
        }

        bw.flush();
        bw.close();
    }
}

---

정답 소스 코드를 확인하시려면 solve url 에서 확인 가능합니다.