백준 1181 자바

제가 좋아하는 정렬 문제입니다.

여기서는 Comparator을 사용하는데 길이순, 사전순, 혹은 점수를 여러 조건에 맞춰서 정렬해야 할 때 사용합니다.

(자세한 건 구글에서 찾으세요..)

조건 1,2번을 Comparator로 구현을 한다면

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

이렇게 나옵니다. 두 객체 o1, o2의 길이를 먼저 비교해서 길이가 같으면 사전 순으로, 그 외에에는 길이가 짧은 것부터(오름차순)으로 정렬 한다는 내용입니다. 반대로 내림차순으로 하고 싶다면 o2.length() - o1.length로 바꾸시면 됩니다.

 

아래는 전체 코드입니다.

import java.io.*;
import java.util.*;
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[] arr = new String[N];

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

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

        System.out.println(arr[0]);

        for (int i = 1; i < arr.length; i++) {
            if(!arr[i].equals(arr[i-1])) {
                System.out.println(arr[i]);
            }
        }

    }
}