자바(Java) - 컬렉션 프레임워크(5) Comparable, Comparator

Comparable<T> 인터페이스?

• 클래스 정렬을 구현할 수 있도록 해주는 인터페이스이며, TreeSet 컬렉션에서 사용해야 합니다.
• 자기 자신과 전달받은 매개변수를 비교하는 인터페이스
• CompareTo() 메소드를 오버라이딩 해서 정렬조건을 구현해야 합니다.

compareTo()

반환 값

1. 양의 정수 : 정렬 조건
2. 음의 정수 : 현 상태 유지
3. 0 : 두 객체가 동일하다는 의미로 Set 특성에 의해 해당 객체가 표시 되지 않습니다. 따라서, 이 경우에는 하위 조건을 추가해줘야 합니다.

🐶예제1

public class CarSort implements Comparable<CarSort> {
    String modelName;
    int modelYear;
    String color;

    public CarSort(String modelName, int modelYear, String color) {
        this.modelName = modelName;
        this.modelYear = modelYear;
        this.color = color;
    }

    @Override
    public int compareTo(CarSort o) {
        if(this.modelYear == o.modelYear) {
            return this.modelName.compareTo(o.modelName); 
        }
        return this.modelYear - o.modelYear; // 앞이 뒤보다 크면 정렬 (오름차순)
    }
}

compareTo()
* 문자열일 경우, 아스키코드로 비교함
* 정렬조건만 넣어주면 실제 정렬은 내부에서 처리해줌(quick sort)

import java.util.TreeSet;

public class Main {
    public static void main(String[] args) {
        TreeSet<CarSort> ts = new TreeSet<>();
        ts.add(new CarSort("테슬라", 2023, "Red"));
        ts.add(new CarSort("코나", 2019, "White"));
        ts.add(new CarSort("EV6", 2021, "White"));
        ts.add(new CarSort("티볼리", 2021, "Blue"));

        for(CarSort e : ts) {
            System.out.println("모델명 : " + e.modelName + " / 연식 : " + e.modelYear + " / 색상 : " + e.color);
        }
    }
}

모델명 : 코나 / 연식 : 2019 / 색상 : White
모델명 : EV6 / 연식 : 2021 / 색상 : White
모델명 : 티볼리 / 연식 : 2021 / 색상 : Blue
모델명 : 테슬라 / 연식 : 2023 / 색상 : Red

Comparator<T> 인터페이스?

• 역시 클래스 정렬을 구현할 수 있도록 해주는 인터페이스이며, TreeSet 컬렉션에서 사용해야 합니다.
• 매개변수에서 전달받은 2개의 객체를 비교하는 인터페이스
• Compare() 메소드를 오버라이딩 해서 정렬조건을 구현해야 합니다.

compare()

반환 값

1. 양의 정수 : 정렬 조건
2. 음의 정수 : 현 상태 유지
3. 0 : 두 객체가 동일하다는 의미로 Set 특성에 의해 해당 객체가 표시 되지 않습니다. 따라서, 이 경우에는 하위 조건을 추가해줘야 합니다.

public class Fruit {
    String name;
    int price;

    public Fruit(String name, int price) {
        this.name = name;
        this.price = price;
    }
}

import java.util.Comparator;

public class DescendComp implements Comparator<Fruit> {
    @Override
    public int compare(Fruit o1, Fruit o2) {
        if(o1.price == o2.price) {
            return o1.name.compareTo(o2.name);
        }
        return o2.price - o1.price; // 큰 값 부터 작은 값 순서로 정렬 (내림차순)
    }

    @Override
    public Comparator<Fruit> reversed() {
        return Comparator.super.reversed();
    }
}

import java.util.TreeSet;

public class Main {
    public static void main(String[] args) {
        TreeSet<Fruit> ts = new TreeSet<>(new DescendComp());
        ts.add(new Fruit("포도", 5500));
        ts.add(new Fruit("사과", 12000));
        ts.add(new Fruit("딸기", 16000));
        ts.add(new Fruit("바나나", 5500));

        for(Fruit e : ts) {
            System.out.println(e.name + " / " + e.price);
        }
    }
}

TreeSet<Fruit> ts = new TreeSet<>(new DescendComp());

객체 생성시 Comparator 정렬 조건을 넣어주는 이유는?
Comparator provided at set creation time, depending on which constructor is used.
- 출처 : Oracle Java SE 11 & JDK 11

🐸실습예제✏️

문제 : 먼저 학생의 수를 입력 받는다. 학생의 수 만큼 이름, 성적, 학번을 입력받아 학생의 성적이 높은 순으로 결과를 표시하고, 성적이 같으면 학번 순으로 정렬하는 프로그램을 작성하라. (출력시 등수도 표시되도록)

 

입력 예시 :

학생 수 : 3

안땡땡 54 202301

김웅니 88 202303

최땡떙 72 202302

 

출력 예시:

====== 학생 성적 출력 ======

------------------------------------------

석차     이름      성적      학번

  1       김웅니     88      202303

  2       최땡떙     72      202302

  3       안땡땡     54      202301

------------------------------------------

public class Student implements Comparable<Student> {
    String name;
    int score;
    String studentNum;

    public Student(String name, int score, String studentNum) {
        this.name = name;
        this.score = score;
        this.studentNum = studentNum;
    }

    @Override
    public int compareTo(Student o) {
        if(this.score == o.score) {
            return this.studentNum.compareTo(o.studentNum);
        }
        return o.score - this.score;
    }
}

import java.util.Scanner;
import java.util.TreeSet;

public class Main {
    public static void main(String[] args) {
        TreeSet<Student> ts = new TreeSet<>();
        Scanner sc = new Scanner(System.in);
        System.out.print("학생 수 : ");
        int num = sc.nextInt();
        for (int i = 0; i < num; i++) {
            String name = sc.next();
            int score = sc.nextInt();
            String studentNum = sc.next();
            ts.add(new Student(name, score, studentNum));
        }
        System.out.println("  < 학생 성적 출력  >  ");
        System.out.println("---------------------------");
        System.out.println("석차    이름    성적    학번");
        System.out.println("---------------------------");
        int grade = 0;
        for (Student e : ts) {
            grade++;
            System.out.println(" " + grade + "    " + e.name + "     " + e.score + "    " + e.studentNum);
        }
    }
}