[Baekjoon 11497] 통나무 건너뛰기 - JAVA

[Silver I] 통나무 건너뛰기 - 11497

문제 링크

성능 요약

메모리: 49752 KB, 시간: 556 ms

분류

그리디 알고리즘, 정렬

문제 설명

남규는 통나무를 세워 놓고 건너뛰기를 좋아한다. 그래서 N개의 통나무를 원형으로 세워 놓고 뛰어놀려고 한다. 남규는 원형으로 인접한 옆 통나무로 건너뛰는데, 이때 각 인접한 통나무의 높이 차가 최소가 되게 하려 한다.

통나무 건너뛰기의 난이도는 인접한 두 통나무 간의 높이의 차의 최댓값으로 결정된다. 높이가 {2, 4, 5, 7, 9}인 통나무들을 세우려 한다고 가정하자. 이를 [2, 9, 7, 4, 5]의 순서로 세웠다면, 가장 첫 통나무와 가장 마지막 통나무 역시 인접해 있다. 즉, 높이가 2인 것과 높이가 5인 것도 서로 인접해 있다. 배열 [2, 9, 7, 4, 5]의 난이도는 |2-9| = 7이다. 우리는 더 나은 배열 [2, 5, 9, 7, 4]를 만들 수 있으며 이 배열의 난이도는 |5-9| = 4이다. 이 배열보다 난이도가 낮은 배열은 만들 수 없으므로 이 배열이 남규가 찾는 답이 된다.

입력

입력은 T개의 테스트 케이스로 이루어져 있다. 첫 줄에 T가 주어진다.

이어지는 각 줄마다 첫 줄에 통나무의 개수를 나타내는 정수 N(5 ≤ N ≤ 10,000), 둘째 줄에 각 통나무의 높이를 나타내는 정수 Li가 주어진다. (1 ≤ Li ≤ 100,000)

출력

각 테스트 케이스마다 한 줄에 주어진 통나무들로 만들 수 있는 최소 난이도를 출력하시오.

문제 풀이

해당 문제는 단순하게 정렬만 하면 안된다.

위 그림처럼 중앙값이 최댓값이 되도록 정렬해야 한다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();

        int T = Integer.parseInt(br.readLine());

        for (int i = 0; i < T; i++) {
            int size = Integer.parseInt(br.readLine());
            int[] numbers = new int[size];
            StringTokenizer st = new StringTokenizer(br.readLine());

            for (int j = 0; j < size; j++) {
                numbers[j] = Integer.parseInt(st.nextToken());
            }

            Arrays.sort(numbers);

            int[] temp = new int[size];
            for (int j = 0; j < size; j = j + 2) {
                if (size % 2 == 1) {
                    if (j == size - 1) {
                        temp[size / 2] = numbers[j];
                    } else {
                        temp[j / 2] = numbers[j];
                        temp[size - 1 - j / 2] = numbers[j + 1];
                    }
                } else {
                    temp[j / 2] = numbers[j];
                    temp[size - 1 - j / 2] = numbers[j + 1];
                }
            }

            int maxDifference = Math.abs(temp[0] - temp[size - 1]);
            for (int j = 0; j < size - 1; j++) {
                int difference = Math.abs(temp[j] - temp[j + 1]);
                if (difference > maxDifference) {
                    maxDifference = difference;
                }
            }

            sb.append(maxDifference).append("\n");
        }

        System.out.println(sb.toString());
    }
}

코드 풀이

        int T = Integer.parseInt(br.readLine());

• 테스트 케이스의 개수를 저장한다.

        for (int i = 0; i < T; i++) {
            int size = Integer.parseInt(br.readLine());
            int[] numbers = new int[size];
            StringTokenizer st = new StringTokenizer(br.readLine());

• 각 테스트 케이스에 대해 반복문을 수행한다.
• 배열의 크기를 읽어온다.
• 숫자를 저장할 배열을 생성한다.
• StringTokenizer를 사용하여 공백을 기준으로 숫자들을 분리한다.

            for (int j = 0; j < size; j++) {
                numbers[j] = Integer.parseInt(st.nextToken());
            }

• 분리된 숫자들을 배열에 저장한다.

            Arrays.sort(numbers);

• 배열을 오름차순으로 정렬한다.

            int[] temp = new int[size];
            for (int j = 0; j < size; j = j + 2) {
                if (size % 2 == 1) {
                    if (j == size - 1) {
                        temp[size / 2] = numbers[j];
                    } else {
                        temp[j / 2] = numbers[j];
                        temp[size - 1 - j / 2] = numbers[j + 1];
                    }
                } else {
                    temp[j / 2] = numbers[j];
                    temp[size - 1 - j / 2] = numbers[j + 1];
                }
            }

• 재배열된 숫자를 저장할 임시 배열 temp를 생성한다.
• size가 홀수인 경우와 짝수인 경우에 따라 숫자를 재배열한다.
• 재배열된 숫자들을 temp 배열에 저장한다.

            int maxDifference = Math.abs(temp[0] - temp[size - 1]);
            for (int j = 0; j < size - 1; j++) {
                int difference = Math.abs(temp[j] - temp[j + 1]);
                if (difference > maxDifference) {
                    maxDifference = difference;
                }
            }

• 재배열된 배열에서 인접한 숫자들의 최대 차이를 계산한다.

            sb.append(maxDifference).append("\n");
        }

• 각 테스트 케이스의 최대 차이를 결과 문자열 sb에 추가한다.

        System.out.println(sb.toString());
    }
}

• 결과를 출력한다.