Java 기초 - 컬렉션 프레임워크 (7) [Comparable과 Comparator]

Comparable<T> 인터페이스

• Comparable인터페이스는 객체를 정렬하는 데 사용되는 메소드인 compareTo()메소드를 정의하고 있습니다.
• 자바에서 같은 타입의 인스턴스를 서로 비교해야만 하는 클래스들은 모두 Comparable인터페이스를 구현하고 있습니다.
• 따라서 Boolean을 제외한 래퍼 클래스나 String, Time, Date와 같은 클래스의 인스턴스는 모두 정렬 가능합니다.
• 이때 기본 정렬 순서는 작은 값에서 큰 값으로 정렬되는 오름차순이 됩니다.

다음 예제는 인스턴스의 비교를 위해 사용자 정의 클래스인 Car 클래스가 Comparable인터페이스를 구현하는 예제입니다.

class Car {
    private String modelName;
    private int modelYear;
    private String color;

    Car(String mn, int my, String c) {
        this.modelName = mn;
        this.modelYear = my;
        this.color = c;
    }

    public int getModelYear() {
        return this.modelYear;
    }

    public String getModel() {
        return this.modelYear + "식 " + this.modelName + " " + this.color;
    }

    public int compareTo(Car obj, Comparable<Car> carComparable) {
        System.out.println(carComparable.compareTo(obj));
    }
}

public class Comparable01 {
    public static void main(String[] args) {
        Car car01 = new Car("아반떼", 2016, "노란색");
        Car car02 = new Car("소나타", 2010, "흰색");

        car01.compareTo(car02, (obj) -> {
            if (car01.getModelYear() == obj.getModelYear()) {
                return 0;
            } else if(car01.getModelYear() < obj.getModelYear()) {
                return -1;
            } else {
                return 1;
            }
        });
    }
}

class Car implements Comparable<Car> {
    private String modelName;
    private int modelYear;
    private String color;

    Car(String mn, int my, String c) {
        this.modelName = mn;
        this.modelYear = my;
        this.color = c;
    }

    public String getModel() {
        return this.modelYear + "식 " + this.modelName + " " + this.color;
    }

    public int compareTo(Car obj) {
        if (this.modelYear == obj.modelYear) {
            return 0;
        } else if(this.modelYear < obj.modelYear) {
            return -1;
        } else {
            return 1;
        }

    }
}

public class Comparable01 {
    public static void main(String[] args) {
        Car car01 = new Car("아반떼", 2016, "노란색");
        Car car02 = new Car("소나타", 2010, "흰색");

        System.out.println(car01.compareTo(car02));
    }
}

실행 결과

Comparable인터페이스는 다음과 같은 메소드를 사용하여 객체를 정렬합니다.

메소드	설명
int compareTo(T o)	해당 객체와 전달된 객체의 순서를 비교함.

Comparator<T> 인터페이스

• Comparable인터페이스는 Comparable인터페이스와 같이 객체를 정렬하는 데 사용되는 인터페이스입니다.
• Comparable인터페이스를 구현한 클래스는 기본적으로 오름차순으로 정렬됩니다.
• 반면에 Comparator인터페이스는 내림차순이나 아니면 다른 기준으로 정렬하고 싶을 때 사용할 수 있습니다.
• 즉, Comparator인터페이스를 구현하면 오름차순 이외의 기준으로도 정렬할 수 있게 되는 것입니다.
• 이때 Comparator인터페이스를 구현한 클래스에서는 compare()메소드를 재정의하여 사용하게 됩니다.

다음 예제는 요소를 내림차순으로 정렬하여 저장하는 TreeSet 인스턴스를 생성하기 위해 Comparator인터페이스를 구현하는 예제입니다.

public class Comparable02 {
	public static void main(String[] args) {
		TreeSet<Integer> ts = new TreeSet<>((o1, o2) -> {
		    if(o1 instanceof Comparable && o2 instanceof Comparable) {
		        Integer c1 = o1;
		        Integer c2 = o2;
		
		        return c2.compareTo(c1);
		    }
		    return -1;
		});
		
		ts.add(30);
		ts.add(40);
		ts.add(20);
		ts.add(10);
		
		Iterator<Integer> iter = ts.iterator();
		while(iter.hasNext()) {
		    System.out.println(iter.next());
		}
	}
}

import java.util.*;

class DescendingOrder implements Comparator<Integer> {
    public int compare(Integer o1, Integer o2) {
        if(o1 instanceof Comparable && o2 instanceof Comparable) {
            Integer c1 = (Integer)o1;
            Integer c2 = (Integer)o2;
            return c2.compareTo(c1);
        }
        return -1;
    }
}

public class Comparable02 {
    public static void main(String[] args) {
        TreeSet<Integer> ts = new TreeSet<Integer>(new DescendingOrder());

        ts.add(30);
        ts.add(40);
        ts.add(20);
        ts.add(10);

        Iterator<Integer> iter = ts.iterator();
        while(iter.hasNext()) {
           System.out.println(iter.next());
        }	
    }
}

실행 결과

Comparator인터페이스는 다음과 같은 메소드를 사용하여 객체를 정렬합니다.

메소드	설명
int compare(T o1, T o2)	전달된 두 객체의 순서를 비교함.
boolean equals(Object obj)	해당 comparator와 전달된 객체가 같은지를 확인함.
default Comparator<T> reversed()	해당 comparator의 역순인 comparator를 반환함.

---

[참고 자료]

http://www.tcpschool.com/java/java_collectionFramework_comparable