[Java] 백준 1181 - 단어 정렬

N 개의 단어를 받아서

1. 길이가 짧은 것부터

2. 길이가 같으면 사전 순으로

정렬하여 출력하는 문제입니다.

하지만 문제를 자세히 읽어보면 중복된 단어의 경우 한번만 출력한다는 조건이 하나 더 있습니다.

그래서 입력받은 문자를 Set 자료구조에 담아서 중복을 제거한 뒤에 ArrayList로 옮겨서 정렬해주었습니다.

import java.util.*;

public class Main {

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int N = Integer.parseInt(sc.nextLine());

        // 중복 제거를 위해 set으로 먼저 input
        Set<String> set = new HashSet<>();

        for (int i = 0; i < N; i++){
            set.add(sc.nextLine());
        }

        // List 변환
        ArrayList<String> list = new ArrayList<>(set);

        // Comparator 클래스를 통하여 custom 정렬
        // 길이에 따라서 먼저 정렬하고 길이가 같으면 사전순으로 정렬
        Collections.sort(list, new Comparator<String>() {
            public int compare(String v1, String v2) {
                if(v1.length() > v2.length())
                    return 1;
                else if(v1.length() < v2.length())
                    return -1;
                else
                    return v1.compareTo(v2);
            }
        });

        for(String s : list)
            System.out.println(s);
    }
    }