[Baekjoon] 6235_Argus

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

---

< Argus >

 

문제 풀이 

 

우선순위 큐를 사용하여 결과를 반환하는 순서를 구한다. 

 

my solution (Java)

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.Comparator;
import java.util.PriorityQueue;
import java.util.StringTokenizer;

public class _6235_ { // Argus

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

		String str = "";
		PriorityQueue<int[]> queue = new PriorityQueue<>(new Comparator<int[]>() {

			@Override
			public int compare(int[] o1, int[] o2) {
				if (o1[1] == o2[1]) {
					return o1[0] - o2[0];
				}
				return o1[1] - o2[1];
			}
		});

		while (!(str = bf.readLine()).equals("#")) {
			st = new StringTokenizer(str);

			st.nextToken();
			int num = Integer.parseInt(st.nextToken());
			int p = Integer.parseInt(st.nextToken());
			queue.add(new int[] { num, p, p });
		}

		int k = Integer.parseInt(bf.readLine());

		for (int i = 0; i < k; i++) {
			int temp[] = queue.poll();
			bw.write(temp[0] + "\n");

			queue.add(new int[] { temp[0], temp[1] + temp[2], temp[2] });
		}
		bw.flush();
	}
}

변수)
str : 입력값
queue : 우선순위 큐 
num, p : 질의의 ID 번호, 결과를 반환하는 주기 
k : 출력해야 하는 개수 

 

#가 입력될 때까지 입력받아 [질의의 ID 번호, 시간, 결과를 반환하는 주기]를 우선순위 큐에 저장한다. k를 입력받는다. k만큼 우선순위 큐에서 poll 하여 값을 출력하고, 다시 우선순위큐에 시간을 더해 저장한다.   

---

---