[Java] 백준-1181번 "단어 정렬"

문제

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

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

---

입력

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

---

출력

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

---

예제1

입력:
13
but
i
wont
hesitate
no
more
no
more
it
cannot
wait
im
yours

출력: 
i
im
it
no
but
more
wait
wont
yours
cannot
hesitate

---

알고리즘

[Java] 백준-11650번 "좌표 정렬하기" (바로가기)
"좌표 정렬하기"에서 사용했었던 Comparator를 이용한다.

먼저 길이를 비교하고, 동일하면 사전 순으로 정렬한다.
동일하지 않으면 바로 정렬해준다.

그 다음, 이전에 동일한 단어가 있었는지 확인하여 없었다면 string buffer에 append해준다.

---

내 소스 코드

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

public class Main {
    public static void main(String[] args) {
        Scanner s = new Scanner(System.in);
        int n=0, cnt=0;
        StringBuffer sb = new StringBuffer();

        n = s.nextInt();

        String[] arr = new String[n];
        for(int i=0;i<n;i++) {
          arr[i] = s.next();
        }

        Arrays.sort(arr, new Comparator<String>() {		
          @Override
          public int compare(String s1, String s2) {
            if(s1.length() == s2.length()) return s1.compareTo(s2);
            else return s1.length() - s2.length();
          }
        });

        for(int i=0;i<n;i++) {
          for(int j=0;j<i;j++) {
            if(arr[i].equals(arr[j])) cnt++;
          }

          if(cnt == 0) {
            sb.append(arr[i]);
            sb.append("\n");
          }
          
          cnt = 0;
        }

        System.out.print(sb);
    }
}