[백준][JAVA]1181번 : 단어 정렬

 

---

먼저 배열에 모든 원소를 입력받아 저장하고 sort() 를 사용하여 위에서 배웠던 알고리즘을 통해 정렬을 한다.

그러면 기본적으로 길이 순으로 정렬이 되고 길이가 같을 경우는 사전 순으로 정렬이 된다.

 

이때 출력할 때에는 정렬된 배열에서 출력하려는 문자열이 직전 배열에 있는 문자열과 같지 않을 경우만 출력하여 문제의 조건대로 '중복되는 문자열은 한 번만 출력'하도록 한다.

 

 

import java.util.*;

class Main {
    public static void main(String args[]) throws Exception {
        Scanner kb = new Scanner(System.in);
        int N=kb.nextInt();
        String[] arr=new String[N];
        kb.nextLine(); // 개행 버리기

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

        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();
                }
            }
        });
        System.out.println(arr[0]);
        for(int i=1; i<N; i++){
            // 중복 피하기
            if(!arr[i].equals(arr[i-1])){
                System.out.println(arr[i]);
            }
        }
    }
}