백준 2822 점수 계산

문제

방법

정답률이 70%여서 너무 쉬운 문제인 줄 알았지만, 약간의 생각이 필요한 문제였다.

score 값과 idx 값을 함께 다루기 위해서 People이라는 Model을 사용했다.

그러고 나서 우선순위큐로 풀면 쉽게 풀 수 있을 것 같아서 사용하였는데, 기본적으로 우선순위큐를 poll할 때에는 작은 값부터 빼내기 때문에 처음부터 넣을때 reverse해서 넣도록 처리했다. 

조금 헷갈렸던 부분은 idx는 큰 값이 있는 위치인 줄 알고 idx를 따로 저장했었다가, 알고 보니 정답은 오름차순으로 출력하면 되어서 동일하게 queue2로 우선순위큐로 선언해서 사용했다. 

 

코드

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;

public class Main {
    // BOK_2822

    static class People {
        int score;
        int idx;

        People(int score, int idx) {
            this.score = score;
            this.idx = idx;
        }
    }
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        PriorityQueue<People> queue = new PriorityQueue<>(
            Comparator.comparingInt((People p) -> p.score).reversed()
        );
        for (int i = 1; i <= 8; i++) {
            int num = Integer.parseInt(br.readLine());
            queue.add(new People(num, i));
        }

        int result = 0;
        PriorityQueue<Integer> queue2 = new PriorityQueue<>();
        for (int i = 0; i < 5; i++) {
            People people = queue.poll();
            result += people.score;
            queue2.add(people.idx);
        }

        bw.write(result +"\n");
        while(!queue2.isEmpty()) {
            bw.write(queue2.poll() + " ");
        }

        bw.flush();
        bw.close();
    }
}