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

영단어 암기는 괴로워  

 

문제

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

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

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

입력

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

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

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

출력

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

 

 

---

예제 입력 출력 예시

예제 입력 1 복사

7 4
apple
ant
sand
apple
append
sand
sand

예제 출력 1 복사

sand
apple
append

예제 입력 2 복사

12 5
appearance
append
attendance
swim
swift
swift
swift
mouse
wallet
mouse
ice
age

예제 출력 2 복사

swift
mouse
appearance
attendance
append
wallet

 

 

문제 풀이

문제 정리

• 자주 나오는 단어를 앞에 배치하여 빈도수를 파악한다.
• 단어의 길이에 따라 배치가 다르기 떄문에 단어 길이를 체크한다.
• 매개변수 m보다 짧은 단어는 체크하지 않는다.

 

 

코드 분석

 

1. 단어의 종류와 빈도수를 저장할 HashMap<String,Integer> 생성
2. 단어의 길이가 매개변수 m보다 작다면 넘어간다.
3. getOrDefault를 이용하여 map에 저장할 때 key 값이 있다면 해당 key에 대한 value 값을 가져오며 key가 없다면 설정한 기본값 0을 가져온다.

HashMap<String,Integer> map = new HashMap<>();
	int n = Integer.parseInt(st.nextToken()); // 단어의 수
	int m = Integer.parseInt(st.nextToken()); // 외울 단어의 길이 기준
	
	for(int i=0; i<n;i++) {
		String word = br.readLine();
		if(word.length()<m) continue;
			map.put(word, map.getOrDefault(word, 0)+1);
	}

 

 

4. Map에 있는 값을 다 꺼내서 list에 저장
5. 저장한 list를 이제 순서대로 바꾼다.
5-1. 빈도수가 같다면 사전순으로, 자주 나오는 단어를 앞에 배치

List<String> list = map.keySet().stream().collect(Collectors.toList());
		
		list.sort((o1,o2)->{
			int a= map.get(o1);
			int b= map.get(o2);
			
			if(a==b) {
				if(o1.length()==o2.length()) {
					return o1.compareTo(o2); //사전순
				}
				return o2.length()-o1.length(); // rl
			}
			return b-a; // 자주 나오는 단어이면 앞에 배치
		});

 

 

 

최종 코드

package baekjoon;

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.HashMap;
import java.util.List;
import java.util.StringTokenizer;
import java.util.stream.Collectors;

public class Main {
	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());
		HashMap<String,Integer> map = new HashMap<>();
		int n = Integer.parseInt(st.nextToken()); // 단어의 수
		int m = Integer.parseInt(st.nextToken()); // 외울 단어의 길이 기준
		
		for(int i=0; i<n;i++) {
			String word = br.readLine();
			if(word.length()<m) continue;

				map.put(word, map.getOrDefault(word, 0)+1);

		}
		// keySet() : map에 있는 값 다 꺼내기 key=value
		List<String> list = map.keySet().stream().collect(Collectors.toList());
		
		list.sort((o1,o2)->{
			int a= map.get(o1);
			int b= map.get(o2);
			
			if(a==b) {
				if(o1.length()==o2.length()) {
					return o1.compareTo(o2); //사전순
				}
				return o2.length()-o1.length(); // rl
			}
			return b-a; // 자주 나오는 단어이면 앞에 배치
		});
		
		for(String words : list) {
			bw.write(words + "\n");
		}
		bw.flush();
        bw.close();

	}
}