[Java] 백준 20920 영단어 암기는 괴로워

https://www.acmicpc.net/problem/20920

난이도

실버3

문제

화은이는 이번 영어 시험에서 틀린 문제를 바탕으로 영어 단어 암기를 하려고 한다. 그 과정에서 효율적으로 영어 단어를 외우기 위해 영어 단어장을 만들려 하고 있다. 화은이가 만들고자 하는 단어장의 단어 순서는 다음과 같은 우선순위를 차례로 적용하여 만들어진다.

1. 자주 나오는 단어일수록 앞에 배치한다.
2. 해당 단어의 길이가 길수록 앞에 배치한다.
3. 알파벳 사전 순으로 앞에 있는 단어일수록 앞에 배치한다

$M$보다 짧은 길이의 단어의 경우 읽는 것만으로도 외울 수 있기 때문에 길이가
$M$이상인 단어들만 외운다고 한다. 화은이가 괴로운 영단어 암기를 효율적으로 할 수 있도록 단어장을 만들어 주자.

입력

첫째 줄에는 영어 지문에 나오는 단어의 개수 $N$과 외울 단어의 길이 기준이 되는 $M$이 공백으로 구분되어 주어진다. ($1 \leq N \leq 100\,000$, $1 \leq M \leq 10$)

둘째 줄부터
$N+1$번째 줄까지 외울 단어를 입력받는다. 이때의 입력은 알파벳 소문자로만 주어지며 단어의 길이는
$10$을 넘지 않는다.

단어장에 단어가 반드시 1개 이상 존재하는 입력만 주어진다.

출력

화은이의 단어장에 들어 있는 단어를 단어장의 앞에 위치한 단어부터 한 줄에 한 단어씩 순서대로 출력한다.

내 코드

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

class Main {
	public static class Word implements Comparable<Word> {
		String word;
		int len, cnt;

		public Word(String word, int len, int cnt) {
			this.word = word;
			this.len = len;
			this.cnt = cnt;
		}

		@Override
		public int compareTo(Word o) {
			if (this.cnt != o.cnt) {
				return Integer.compare(o.cnt, this.cnt);
			} else if (this.len != o.len) {
				return Integer.compare(o.len, this.len);
			}
			return this.word.compareTo(o.word);
		}
	}

	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine());
		int N = Integer.parseInt(st.nextToken());
		int M = Integer.parseInt(st.nextToken());

		HashMap<String, Integer> set = new HashMap<>();
		for (int i = 0; i < N; i++) {
			String w = br.readLine();
			if (w.length() >= M) {
				if (set.containsKey(w)) {
					set.replace(w, set.get(w) + 1);
				} else {
					set.put(w, 0);
				}
			}
		}

		PriorityQueue<Word> pq = new PriorityQueue<>();
		for (String w : set.keySet()) {
			pq.add(new Word(w, w.length(), set.get(w)));
		}

		StringBuilder sb = new StringBuilder();
		while (!pq.isEmpty()) {
			sb.append(pq.poll().word).append("\n");
		}
		
		System.out.println(sb);
	}
}

주어진 단어의 길이가 M 이상인 경우에만
HashMap으로 이미 나왔는지 지금까지 몇 번 나왔는지 확인하며 저장하였다.
그리고 우선순위큐에 단어, 길이, 카운트 한 숫자를 담은 Word 클래스를 넣어서
출력하면 되는 문제이다.

근데 시간초과가 나서 StringBuilder로 담은 다음에 출력을하니 맞았다.
긴 답안을 출력할 때는 StringBuilder로 담아야겠다.

요즘 푼 문제여도 다시 풀어보는데 풀 때마다 코드 방향이 달라서 신기하다,,
시간과 메모리가 줄어들었는지 늘어났는지 확인하며
어떤 자료구조를 쓰는게 좋은지 비교하며 또 새롭게 배우는 것 같다.