[Baekjoon] 1461_도서관

Gold IV

문제(출처: https://www.acmicpc.net/problem/1461)

---

< 도서관 >

 

문제 풀이 

 

마지막에는 다시 0으로 돌아올 필요가 없기 때문에 가장 먼 곳을 마지막에 가야 한다. 가장 먼 곳을 제외하고 먼 곳을 순서대로 가져다 놓는다.

 

 my solution (Java)

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Collections;
import java.util.StringTokenizer;

public class _1461_ { // 도서관

	public static void main(String[] args) throws IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(bf.readLine());

		int n = Integer.parseInt(st.nextToken());
		int m = Integer.parseInt(st.nextToken());

		ArrayList<Integer> plus = new ArrayList<>();
		ArrayList<Integer> minus = new ArrayList<>();

		st = new StringTokenizer(bf.readLine());
		for (int i = 0; i < n; i++) {
			int num = Integer.parseInt(st.nextToken());
			if (num < 0) {
				minus.add(num);
			} else {
				plus.add(num);
			}
		}

		Collections.sort(plus);
		Collections.sort(minus);

		int result = 0;

		int idx = plus.size() - 1;
		while (idx >= 0) {
			result += plus.get(idx) * 2;
			idx -= m;
		}

		idx = 0;
		while (idx < minus.size()) {
			result += (-minus.get(idx)) * 2;
			idx += m;
		}

		if (plus.size() > 0 && minus.size() > 0) {
			if (-minus.get(0) > plus.get(plus.size() - 1)) {
				result += minus.get(0);
			} else {
				result -= plus.get(plus.size() - 1);
			}
		}else {
			if(plus.size()>0) {
				result -= plus.get(plus.size() - 1);
			}else {
				result += minus.get(0);
			}
		}

		System.out.println(result);
	}
}

변수)
n, m : 책의 개수, 한 번에 들 수 있는 책의 개수
plus, minus : 책의 위치
result : 최소 걸음 수

 

책의 개수, 한 번에 들 수 있는 책의 개수를 입력받는다. 책의 위치를 입력받으면서 -, +를 구분해서 저장한다. 각각 ArrayList를 오름차순으로 정렬한다. +를 저장한 ArrayList는 뒤에서부터 m 개씩 제자리에 놔둔다. -를 저장한 ArrayList는 앞에서부터 m 개씩 제자리에 놔둔다. 가장 먼 곳에 책을 제자리에 둔 후에는 다시 0으로 돌아올 필요 없으므로 +, -를 저장한 ArrayList를 보며 가장 먼 위치를 확인한 후 그 값을 빼준다. 

---

---