정렬: 단어 정렬 (백준 1181)

○ 문제

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

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

단, 중복된 단어는 하나만 남기고 제거해야 한다.

 

○ 입력

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

 

○ 출력

조건에 따라 정렬하여 단어들을 출력한다.

 

○ 예제 입력

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

---

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.io.IOException;
import java.util.Arrays;
import java.util.Comparator;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        
        int num = Integer.parseInt(br.readLine());
        String[] arr = new String[num];
        
        for (int i = 0; i < num; i++) {
            arr[i] = br.readLine();
        }
        
        //중복제거
        String[] newArr = Arrays.stream(arr).distinct().toArray(String[]::new);
        
        //정렬
        Arrays.sort(newArr, new Comparator<String>() {
            @Override
            public int compare(String s1, String s2) {
                int result = s1.length() - s2.length(); //길이
                if (result == 0) {
                    result = s1.compareTo(s2);  //알파벳순
                }
                
                return result;
            }
        });
        
        for (String s : newArr) {
            bw.write(s+"\n");
        }
        
        bw.flush();
        bw.close();
        
    }
}

○ 확인

- compareTo() 정수, 문자열 인자 두 개 비교하여 정수 리턴

- int compareTo(NumberSubClass referenceName)

- int compareTo(String anotherString)

- 숫자의 비교 같은 경우는 단순히 크다(1), 같다(0), 작다(-1)

- 문자열의 비교 같은 경우는 같다(0), 그 외 양수/음수값

- Arrays.sort 내에서 compare 메서드 override