[BaekJoon 1181번 / JAVA] 단어 정렬

Question

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

1181번: 단어 정렬

---

💡 Solution

HashMap을 key나 value에 따라 정렬할 수 있다면, 쉽게 해결할 수 있는 문제였습니다.

 

💻 Code

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

public class BJ1181 {

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

		StringBuilder sb = new StringBuilder();
		HashMap<String, Integer> map = new HashMap();
		for (int i = 0; i < N; i++) {
			String str = br.readLine();
			map.put(str, str.length());
		}

		//map을 길이 값 기준으로 오름차순 정렬
		//길이가 동일할 경우 키를 기준으로 오름차순(사전순) 정렬
		List<Map.Entry<String, Integer>> entries = map.entrySet().stream()
				.sorted(Map.Entry.<String, Integer>comparingByValue()
						.thenComparing(Map.Entry.comparingByKey()))
				.collect(Collectors.toList());
		
		for (Map.Entry<String, Integer> entry : entries) {
			sb.append(entry.getKey()).append("\n");
		}
		
		System.out.print(sb);
	}

}