글 읽기 - 어디가 틀렸는지 모르겠습니다. ㅠㅠ

smr603 4년 전 0

알고리즘 초보입니다.

며칠째 잡고있는데 도통 어디가 틀렸는지 모르겠습니다. 고수님들 확인한번 부탁드립니다. 감사합니다.

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

public class Main {
	
	static class Student implements Comparable<Student>{
		int index;
		int number;
		int score;
		
		Student(int index, int number, int score) { 
			this.index = index;
			this.number = number;
			this.score = score; 
		}
	
		@Override
		public int compareTo(Student s) {
			if (this.score > s.score) { // 점수가 가장 적은게 맨 앞으로 (삭제대상)
				return 1;
			} else if(this.score == s.score) {
				if (this.index > s.index) { // 점수가 같다면 오래된 게시물이 앞으로 (삭제대상)
					return 1;
				}
			}
			return -1;
		}
	}
	
	static int n, k;
	static Student[] students = new Student[101];
	static PriorityQueue<Student> pq = new PriorityQueue<>();
	
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader( System.in ));
		n = Integer.parseInt(br.readLine());
		k = Integer.parseInt(br.readLine());
		
		StringTokenizer st = new StringTokenizer(br.readLine());
		for (int i = 0; i < k; i++) {
			int num = Integer.parseInt(st.nextToken());
			
			// pq 안에 같은게 있는지 확인.
			if (students[num] != null) { // 이미 사진틀안에 있다면,
				students[num].score ++;
			} else {
				//Student 생성 후 배열에 체크.
				students[num] = new Student(i, num, 1);
				// pq 가 가질 수 있는 최대 길이 n
				if (pq.size() < n) {
					pq.add(students[num]);
				} else {
					// 삭제 후 추가;
					students[pq.poll().number] = null;
					pq.add(students[num]);
				}
			}
		}
		
		pq.stream()
			.mapToInt(s -> s.number)
			.sorted()
			.forEach(n -> System.out.print(n + " "));
	}

}

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

어디가 틀렸는지 모르겠습니다. ㅠㅠ

smr603 4년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말