정렬 문제 해결을 위한 Comparator / Comparable

1431번: 시리얼 번호

 

위의 문제의 경우에는 세 가지의 정렬 기준으로 주어진 입력을 정렬해야 하는데

반복문을 돌면서 총 3번의 정렬을 수행할 수도 있겠지만

Comparator와 Comparable을 이용해 커스텀 정렬 기준으로 정렬을 할 수도 있다.

 

자바에서 제공하는 두 인터페이스들은 객체를 비교할 수 있는 상태로 만들기 위해 사용하는데

숫자나 문자 같은 경우는 기본적으로 손쉽게 비교가 가능했지만

객체는 서로 다른 특성을 가지고 있기 때문에 어떤 것으로 비교해야 할지

정해진 기준이 없기 때문에 두 인터페이스를 구현하여 객체를 어떻게 비교할지 정해주는 것이다.

 

Comparable

public class People implements Comparable<People> {

    int age;
    String name;

    public People(int age, String name) {
        this.age = age;
        this.name = name;
    }

    @Override
    public int compareTo(People o) {
        return this.age - o.age;
    }
}

 

우선 Comparable을 먼저 살펴보자면 구현은 위와 같이 할 수 있다.

 

객체를 어떤 기준으로 비교할지 compareTo 메서드를 구현하여 정의할 수 있는데

메서드의 파라미터로는 비교할 객체를 받아온다.

 

PeopleA와 비교하고자 하는 PeopleB를 파라미터로 받아오면 되는데

즉, 자기 자신과 비교하고자 하는 같은 타입의 상대 객체를 받아오면 된다.

 

compareTo 메서드의 리턴값에 따라 객체의 비교 결과가 달라지는데

양수인 경우에는 자기 자신이 더 큰 경우이고 0인 경우에는 같은 경우

음수인 경우에는 상대 객체가 더 큰 경우이다.

 

Comparator

public class CustomStringComparator implements Comparator<String> {

    @Override
    public int compare(String o1, String o2) {
        return 0;
    }
}

public class CustomIntegerComparator implements Comparator<Integer> {

    @Override
    public int compare(Integer o1, Integer o2) {
        return 0;
    }
}

 

위의 코드를 보면 알 수 있듯이 Comparator는 compare 메서드를 구현하는데

Comparable과 다른 점은 매개 변수로 비교할 두 값을 받아온다는 것이다.

 

즉, Comparator와 Comparable의 차이는

받아온 두 값을 비교하거나 자기 자신과 다른 객체를 비교하는 것이다.

Arrays.sort(arr, new Comparator<String>() {
    @Override
    public int compare(String o1, String o2) {
        return 0;
    }
});

 

별도의 클래스를 만들지 않고 익명 객체로 사용하거나

위와 같이 정렬 메서드의 정렬 기준으로 사용할 수도 있다.