📌 문제

[1181] 단어 정렬

알파벳 소문자로 이루어진 N개의 단어가 들어오면 아래와 같은 조건에 따라 정렬하는 프로그램을 작성하시오.

• 길이가 짧은 것부터
• 길이가 같으면 사전 순으로
  
  단, 중복된 단어는 하나만 남기고 제거해야 한다.

⬇️ 입력

첫째 줄에 단어의 개수 N이 주어진다. (1 ≤ N ≤ 20,000) 둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어진다. 주어지는 문자열의 길이는 50을 넘지 않는다.

⬆️ 출력

조건에 따라 정렬하여 단어들을 출력한다.

💡 코드

✅ String 타입 배열 arr[]에 단어들을 저장하고, Arrays.sort(arr)를 통해 정렬을 수행한다. 이 때, 정렬 기준이 단순 사전 순이 아닌 단어의 길이 + 사전 순 이므로 조건을 정의하기 위해 Comparator 인터페이스를 사용한다. 기본적으로 길이가 짧은 순으로 정렬하고, 길이가 같은 경우 compareTo() 메서드를 사용하여 사전 순으로 정렬한다.

정렬을 마친 후, 요소들을 출력하는데 이 때 중복된 요소는 제거해줘야 한다. 길이 및 사전 순으로 정렬을 마친 후이기 때문에 중복 요소는 연속으로 위치하므로, 조건 !arr[i].equals(arr[i-1])을 통해 특정 요소가 이전 요소와 같지 않은 경우만 출력해주면 된다.

import java.io.*;
import java.util.*;
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));
		
		int n = Integer.parseInt(br.readLine());
		String[] arr = new String[n];
		for(int i=0;i<n;i++) {
			arr[i] = br.readLine();;
		}
		
		Arrays.sort(arr, new Comparator<String>() {
			@Override
			public int compare(String o1, String o2) {
				if(o1.length() == o2.length()) {
					return o1.compareTo(o2);
				} else {
					return o1.length() - o2.length();
				}
			}
		});
		
		StringBuilder sb = new StringBuilder();
		sb.append(arr[0]).append("\n");
		for(int i=1;i<n;i++) {
			if(!arr[i].equals(arr[i-1])) {
				sb.append(arr[i]).append("\n");
			}
		}
		bw.write(sb + "");
		
		br.close();
		bw.close();
	}
}