[BOJ] 1700 멀티탭 스케줄링 - JAVA

1. 문제

1700번: 멀티탭 스케줄링

기숙사에서 살고 있는 준규는 한 개의 멀티탭을 이용하고 있다. 준규는 키보드, 헤어드라이기, 핸드폰 충전기, 디지털 카메라 충전기 등 여러 개의 전기용품을 사용하면서 어쩔 수 없이 각종 전기용품의 플러그를 뺐다 꽂았다 하는 불편함을 겪고 있다. 그래서 준규는 자신의 생활 패턴을 분석하여, 자기가 사용하고 있는 전기용품의 사용순서를 알아내었고, 이를 기반으로 플러그를 빼는 횟수를 최소화하는 방법을 고안하여 보다 쾌적한 생활환경을 만들려고 한다.

예를 들어 3 구(구멍이 세 개 달린) 멀티탭을 쓸 때, 전기용품의 사용 순서가 아래와 같이 주어진다면,

1.키보드
2.헤어드라이기
3.핸드폰 충전기
4.디지털 카메라 충전기
5.키보드
6.헤어드라이기

키보드, 헤어드라이기, 핸드폰 충전기의 플러그를 순서대로 멀티탭에 꽂은 다음 디지털 카메라 충전기 플러그를 꽂기 전에 핸드폰충전기 플러그를 빼는 것이 최적일 것이므로 플러그는 한 번만 빼면 된다.

---

2. 풀이

priorityQueue를 100칸짜리 배열로 만든다.

각 기기를 사용할 때마다 pq에 add한다.

 

멀티탭 이용하기 위한 우선순위큐는 위의 배열값을 이용해서 정렬한다.

pq는 add나 poll할 때 내부정렬되므로 순서에 주의한다.

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

public class Main {
	public static void main(String[] args) throws IOException {
		BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(in.readLine());
		int N = Integer.parseInt(st.nextToken());
		int K = Integer.parseInt(st.nextToken());

		st = new StringTokenizer(in.readLine());
		int[] order = new int[K];
		PriorityQueue<Integer>[] pqArray = new PriorityQueue[101];
		for (int i = 0; i < K; i++) {
			order[i] = Integer.parseInt(st.nextToken());
			if(pqArray[order[i]] == null) pqArray[order[i]] = new PriorityQueue<>();
			pqArray[order[i]].add(i);
		}

		int answer = 0;
		PriorityQueue<Integer> queue = new PriorityQueue<>((o1,o2)->{
			if(pqArray[o1].size()==0 || pqArray[o2].size()==0) return pqArray[o1].size()-pqArray[o2].size();
			return pqArray[o2].peek()-pqArray[o1].peek();
		});

		boolean[] state = new boolean[101];
		for(int i=0;i<K;i++){
			if(state[order[i]]) {
				pqArray[order[i]].poll();
				queue.add(queue.poll());//poll해서 순서때문에 다시해줘야함
				continue;
			}
			if(queue.size()>=N){
				state[queue.poll()] = false;
				answer++;
			}
			pqArray[order[i]].poll();
			queue.add(order[i]);//순서 아주 중요
			state[order[i]] = true;
		}

		System.out.println(answer);

	}

}

---

3. 결과