본문 바로가기
Problem Solving/Baekjoon

[백준] 7662 이중 우선순위 큐 - Data Structure / Java

by graycode 2023. 4. 23.

 문제 링크

 

7662번: 이중 우선순위 큐

입력 데이터는 표준입력을 사용한다. 입력은 T개의 테스트 데이터로 구성된다. 입력의 첫 번째 줄에는 입력 데이터의 수를 나타내는 정수 T가 주어진다. 각 테스트 데이터의 첫째 줄에는 Q에 적

www.acmicpc.net

 

 풀이 과정

 

 풀이 코드

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

public class Main {

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

        StringBuilder sb = new StringBuilder();
        int t = Integer.parseInt(br.readLine());

        while (t-- > 0) {
            TreeMap<Integer, Integer> map = new TreeMap<>();
            int k = Integer.parseInt(br.readLine());

            while (k-- > 0) {
                st = new StringTokenizer(br.readLine());
                String op = st.nextToken();
                int n = Integer.parseInt(st.nextToken());

                if (op.equals("I"))
                    map.put(n, map.getOrDefault(n, 0) + 1);
                else {
                    if (map.isEmpty())
                        continue;

                    n = n == 1 ? map.lastKey() : map.firstKey();
                    if (map.get(n) == 1)
                        map.remove(n);
                    else
                        map.put(n, map.get(n) - 1);
                }
            }

            if (map.isEmpty())
                sb.append("EMPTY\n");
            else
                sb.append(map.lastKey()).append(" ").append(map.firstKey()).append("\n");
        }

        bw.write(sb.toString());
        bw.flush();
    }

}

댓글