[SWEA] 7701. 염라대왕의 이름 정렬

문제

---

String 을 입력받아 중복을 제거하고 사전 순으로 나열하는 문제이다. 중복 제거와 sort 는 기존 라이브러리를 사용할 수 있기 때문에 매우 간단한 문제이다.

실수

---

단순하게 sort 를 사용하면 될 줄 알았지만 문제 조건에 길이가 짧은 것이 더 큰 우선순위를 가졌다. [aaa, ab] 를 입력하면 일반 sort 시 그대로이지만 길이가 짧은 ab가 우선되어야한다. 따라서 sort 시 Compare 함수를 재정의해줬다.

코드

---

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.StringTokenizer;
import java.io.BufferedReader;
import java.io.FileInputStream;
import java.io.InputStreamReader;

public class Solution {
    static StringBuilder result = new StringBuilder();

    public static void main(String args[]) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        int T = Integer.parseInt(br.readLine());

        int N;
        //////////////////////////
        Set<String> names = new HashSet<>();
        ///////////////////////////
        for (int test_case = 1; test_case <= T; test_case++) {
            result.append("#" + test_case + "\n");
            N = Integer.parseInt(br.readLine());
            /////////////////////////////
            for (int i = 0; i < N; i++) {
                names.add(br.readLine());
            }
            List<String> sorted = new ArrayList<>(names);
            Collections.sort(sorted, new Comparator<String>() {
                @Override
                public int compare(String s1, String s2) {
                    if (s1.length() == s2.length()) {
                        return s1.compareTo(s2);
                    }
                    return s1.length() - s2.length();
                };
            });

            //////////////////////////////
            for (String name : sorted) {
                result.append(name + "\n");
            }
            names.clear();
            sorted.clear();
        }
        System.out.println(result);
    }
}