[백준 / Java] 20920: 영단어 암기는 괴로워 (자료구조)

난이도: Silver 3

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

 

문제

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

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

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

입력

첫째 줄에는 영어 지문에 나오는 단어의 개수 N과 외울 단어의 길이 기준이 되는 M이 공백으로 구분되어 주어진다.

(1 ≤ N ≤ 100,000, 1 ≤ M ≤ 10)

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

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

출력

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

---

 

TreeMap을 알아야 하고 TreeMap의 키 모음(keySet)을 ArrayList로 바꿔서 정렬할 줄 알아야 한다. 

 

TreeMap은 키가 중복이 되지 않으면서, 정렬을 자동으로 해준다.

만약 키가 문자열이라면 알파벳 순으로 정렬을 자동으로 해준다. 

 

키 모음(keySet)을 ArrayList로 바꾸는 방법은 ArrayList를 만들 때 생성자 매개변수에 keySet을 넘겨주면 된다.

 

그리고 이 ArrayList의 sort 함수를 사용하여 매개변수로 Comparator 인터페이스를 사용해 compare 메서드를 오버라이딩하여 문제에 주어진 우선순위 조건에 맞 된다.

 

 

자료구조는 코드를 보면 더 이해가 잘 될 것이다.

바로 코드를 보자.

 

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.Comparator;
import java.util.StringTokenizer;
import java.util.TreeMap;
 
public class P20920_영단어암기는괴로워 {
    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 = new StringTokenizer(br.readLine());
        
        int N = Integer.parseInt(st.nextToken());    // 영어 지문에 나오는 단어의 개수
        int M = Integer.parseInt(st.nextToken());    // 외울 단어의 길이 기준
        // 단어장 트리맵 => 단어 중복 저장을 방지하고, 알파벳 사전 순으로 저장시키기 위한 용도
        TreeMap<String, Integer> dict = new TreeMap<>();    
        
        // 단어장 초기화
        for (int i = 0; i < N; i++) {
            String word = br.readLine();    // 입력받은 단어
            // 단어의 길이가 외울 단어의 길이보다 짧다면 건너뛰기
            if (word.length() < M) {    
                continue;                
            }
            // 단어장에 해당 단어가 이미 존재한다면
            if (dict.containsKey(word)) {    
                dict.put(word, dict.get(word) + 1);    // 단어 빈도 수 1 증가
            } else {    // 단어장에 해당 단어가 없다면
                dict.put(word,  1);        // 단어 빈도 수를 1로 설정하고 새로 추가
            }
        }
        
        // 단어장에 존재하는 단어들
        // 우선순위 3: 알파벳 사전 순으로 앞에 있는 단어일수록 앞에 배치 => 이미 만족한 상태
        ArrayList<String> words = new ArrayList<>(dict.keySet());
        
        // 단어 순서를 우선순위별로 정렬 
        words.sort(new Comparator<String>() {
            public int compare(String o1, String o2) {
                int freq1 = dict.get(o1);    // 앞 단어의 빈도 수
                int freq2 = dict.get(o2);    // 뒷 단어의 빈도 수
                // 두 단어의 빈도 수가 다르다면
                if (freq1 != freq2) {    
                    // 우선순위 1: 자주 나오는 단어일수록 앞에 배치
                    return freq2 - freq1;    
                } else {    // 두 단어의 빈도 수가 같다면
                    // 우선순위 2: 해당 단어의 길이가 길수록 앞에 배치
                    return o2.length() - o1.length();
                }
            }
        });
        
        // 정렬된 단어들을 차례대로 버퍼에 작성
        for (String word : words) {
            bw.write(word + "\n");
        }
        
        bw.flush();    // 버퍼에 작성되어있는 상태를 그대로 한꺼번에 출력
    }
}