BOJ1461 도서관 java

문제

세준이는 도서관에서 일한다. 도서관의 개방시간이 끝나서 세준이는 사람들이 마구 놓은 책을 다시 가져다 놓아야 한다. 세준이는 현재 0에 있고, 사람들이 마구 놓은 책도 전부 0에 있다. 각 책들의 원래 위치가 주어질 때, 책을 모두 제자리에 놔둘 때 드는 최소 걸음 수를 계산하는 프로그램을 작성하시오. 세준이는 한 걸음에 좌표 1칸씩 가며, 책의 원래 위치는 정수 좌표이다. 책을 모두 제자리에 놔둔 후에는 다시 0으로 돌아올 필요는 없다. 그리고 세준이는 한 번에 최대 M권의 책을 들 수 있다.

입력

첫째 줄에 책의 개수 N과, 세준이가 한 번에 들 수 있는 책의 개수 M이 주어진다. 둘째 줄에는 책의 위치가 주어진다. N과 M은 50보다 작거나 같은 자연수이다. 책의 위치는 0이 아니며, 절댓값은 10,000보다 작거나 같은 정수이다.

출력

첫째 줄에 정답을 출력한다.

풀이법

• 양수와 음수는 따로 움직여지므로 나눠서 풀이하자.
• 음수와 양수의 절대값 중 가장 큰 값을 가진 수를 마지막 순서로 두자.

import java.util.*;
import java.io.*;

public class Main {
    public static void main(String[] args) throws NumberFormatException, IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());
        int answer = 0;
        List<Integer> positiveBooks = new ArrayList<>();
        List<Integer> negativeBooks = new ArrayList<>();

        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < N; i++) {
            int book = Integer.valueOf(st.nextToken());
            if (book >= 0) {
                positiveBooks.add(book);
            } else {
                negativeBooks.add(book);
            }
        }
        Collections.sort(positiveBooks, Collections.reverseOrder());
        Collections.sort(negativeBooks);

        boolean isLastMovePositive = true;
        if(positiveBooks.size() != 0 && negativeBooks.size() != 0) {
            if(positiveBooks.get(0) < Math.abs(negativeBooks.get(0))) {
                isLastMovePositive = false;
            }
        } else if(positiveBooks.size() == 0) {
            isLastMovePositive = false;
        } else {
            isLastMovePositive = true;
        }

        if (isLastMovePositive) {
            answer += getLastMovingDistance(M, positiveBooks);
        } else {
            answer += getLastMovingDistance(M, negativeBooks);
        }

        while (positiveBooks.size() != 0) {
            answer += getMovingDistance(M, positiveBooks);
        }

        while (negativeBooks.size() != 0) {
            answer += getMovingDistance(M, negativeBooks);
        }

        System.out.println(answer);

    }

    private static int getLastMovingDistance(int m, List<Integer> books) {
        int book = Math.abs(books.get(0));
        for (int i = 0; i < m; i++) {
            if (books.size() == 0) break;
            books.remove(0);
        }

        return book;
    }

    private static int getMovingDistance(int m, List<Integer> books) {
        int book = Math.abs(books.get(0));
        for (int i = 0; i < m; i++) {
            if (books.size() == 0) break;
            books.remove(0);
        }

        return book * 2;
    }
}