[Java] 백준 1181 단어 정렬

단어 정렬

분석

• 알파벳 소문자로 이루어진 N개의 단어가 들어오면 아래와 같은 조건에 따라 정렬

• 길이가 짧은 것부터

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

입력

• N: 단어의 개수 (1 ≤ N ≤ 20,000)
• 둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어짐 (길이 <= 50)

출력

• 조건에 따라 정렬하여 단어들을 출력

import java.util.Scanner;
import java.util.Arrays;
import java.util.Comparator;


public class Main {
    public static void main(String[] args) {

        Scanner in = new Scanner(System.in);

        int N = in.nextInt();
        String[] arr = new String[N];

        in.nextLine();

        for (int i = 0; i < N; i++) {
            arr[i] = in.nextLine();
        }

        Arrays.sort(arr, new Comparator<String>() {
            public int compare(String s1, String s2) {

                if (s1.length() == s2.length()) {
                    return s1.compareTo(s2);
                }

                else {
                    return s1.length() - s2.length();
                }
            }
        });


        System.out.println(arr[0]);

        for (int i = 1; i < N; i++) {
            if (!arr[i].equals(arr[i - 1])) {
                System.out.println(arr[i]);
            }
        }
    }

}

• in.nextLine(); 로 개행문자 처리하는 것도 까먹지 말아야겠다...
• Arrays.sort() 메소드를 사용해 문자열 배열을 정렬했고, Comparator<String>을 사용해 문자열을 길이로 먼저 정렬하고, 길이가 같으면 compareTo()를 이용해 사전순으로 정렬했다
• s1.compareTo(s2)는 두 문자열을 사전 순으로 비교

처음에 중복 처리를 까먹어서 틀렸다
처음에 파이썬처럼 생각해서 람다식으로 하려고 했는데 모르겠어서 탈락
찾아본 결과 람다식 활용하면 이런 코드가 된다고 한다

Arrays.sort(arr, (s1, s2) -> {
            if (s1.length() == s2.length()) {
                return s1.compareTo(s2);  // 길이가 같으면 사전순으로 정렬
            } else {
                return s1.length() - s2.length();  // 길이가 다르면 길이순으로 정렬
            }
        });