[Baekjoon] 5462_POI

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

---

< POI >

 

문제 풀이 

 

: 각 문제 점수

: 각 참가자별 획득 점수

: 각 참가자별 푼 문제 수

를 구하여 우선순위 큐를 활용해 점수 > 푼 문제 수 > ID 순으로 정렬한다.

 

 

 my solution (Java)

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.PriorityQueue;
import java.util.StringTokenizer;

public class _5462_ { // POI
	static class Person implements Comparable<Person> {
		private int score;
		private int problem;
		private int id;
        
		public Person(int score, int problem, int id) {
			this.score = score;
			this.problem = problem;
			this.id = id;
		}
        
		@Override
		public int compareTo(Person o) {
			if (this.score == o.score) {
				if (this.problem == o.problem) {
					return this.id - o.id;
				}
				return o.problem - this.problem;
			}
			return o.score - this.score;
		}
	}
    
	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 T = Integer.parseInt(st.nextToken());
		int P = Integer.parseInt(st.nextToken());
        
		int arr[][] = new int[N][T];
		int score[] = new int[T];
		for (int i = 0; i < N; i++) {
			st = new StringTokenizer(bf.readLine());
			for (int j = 0; j < T; j++) {
				arr[i][j] = Integer.parseInt(st.nextToken());
				if (arr[i][j] == 0) {
					score[j] += 1;
				}
			}
		}
        
		PriorityQueue<Person> queue = new PriorityQueue<>();
        
		for (int i = 0; i < N; i++) {
			int num = 0, s = 0;
			for (int j = 0; j < T; j++) {
				if (arr[i][j] == 1) {
					num += 1;
					s += score[j];
				}
			}
			queue.add(new Person(s, num, i + 1));
		}
        
		int rank = 0;
		Person person = null;
		while (!queue.isEmpty()) {
			person = queue.poll();
			if (person.id == P) {
				rank += 1;
				break;
			}
			rank += 1;
		}
		System.out.println((person.score) + " " + rank);
	}
}

변수)
N, T, P : 참가자 수, 문제 수, P ID
arr : 문제 푼 여부
score : 각 문제 점수
queue : 우선순위 큐
rank : P의 등수

 

Person

점수, 푼 문제 수를 변수로 가짐

점수 내림차순, 푼 문제 수 내림차순, ID 오름차순으로 객체를 비교한다.

 

Main

참가자 수, 문제 수, P ID를 입력받는다. 각 참가자의 문제 푼 여부를 입력받아 arr에 저장하며 각 문제를 풀지 못한 사람의 수를 구해 score에 저장한다. arr을 전체 탐색하며 각 참가자별 점수와 푼 문제 수를 구해 우선순위 큐에 Person 객체로 저장한다. 우선순위 큐를 poll 하면서 P를 찾을 때까지 rank를 구한다. 최종 P의 점수와 rank를 출력한다.

---

---