문제

백준 1181번
https://www.acmicpc.net/problem/1181

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

1. 길이가 짧은 것부터
2. 길이가 같으면 사전 순으로

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

조건에 따라 정렬하여 단어들을 출력한다. 단, 같은 단어가 여러 번 입력된 경우에는 한 번씩만 출력한다.

Code

Java

import java.util.*;
import java.io.*;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());  // 단어의 개수
        String[] str = new String[N];  // 단어를 담을 배열
        StringBuilder sb = new StringBuilder();  // 결과를 담을 StringBuilder

        for (int i = 0; i < N; i++) {
            str[i] = br.readLine();
        }

        // 정렬
        Arrays.sort(str, new Comparator<String>() {
            @Override
            public int compare(String o1, String o2) {
                // 길이가 같을 경우
                if (o1.length() == o2.length()) {
                    return o1.compareTo(o2);
                }
                return o1.length() - o2.length();
            }
        });

        String temp = "";  // 이전 단어를 담아둘 변수
        for (String s : str) {
            // 이전 변수와 같지 않을 경우
            if (!temp.equals(s)) {
                sb.append(s).append("\n");
                temp = s;
            }
        }
        System.out.println(sb);
    }
}