Comparator + 람다식 → 커스텀 정렬
1. 기본 구조
Arrays.sort(배열, (a, b) -> {
return 비교값;
});
- 자바에서 배열을 정렬할 때,
Comparator
를 람다식으로 넘겨 정렬 기준을 직접 지정할 수 있다. - Comparator는 객체 타입인 Integer에만 사용할 수 있기 때문에 int[] → Integer[] 변환이 필요할 때가 있다.
1.1 Comparator 람다식 반환값 기준
반환값 | 의미 | 정렬 결과 |
---|---|---|
음수 ( | a가 b보다 작다 | a가 앞에 옴 |
0 | a와 b가 같다 | 순서 유지 |
양수 ( | a가 b보다 크다 | b가 앞에 옴 |
2. 자주 사용하는 커스텀 정렬
2.1 오름차순 정렬 (작은 수 → 큰 수)
Arrays.sort(arr, (a, b) -> a - b);
2.2 내림차순 정렬 (큰 수 → 작은 수)
Arrays.sort(arr, (a, b) -> b - a);
2.3 절댓값 기준 오름차순 정렬
Arrays.sort(arr, (a, b) -> Math.abs(a) - Math.abs(b));
2.4 절댓값 기준 + 값이 같으면 큰 수 우선
Arrays.sort(arr, (a, b) -> {
int diff = Math.abs(a) - Math.abs(b);
if (diff == 0) return b - a; // 절댓값 같으면 큰 수 우선
return diff;
});
3. 정리
목적 | 정렬식 |
---|---|
기본 오름차순 |
|
기본 내림차순 |
|
절댓값 기준 |
|
절댓값 + 값이 같으면 큰 수 우선 |
|
4. List 정렬
4.1 기본 오름차순 정렬
List<Integer> list = Arrays.asList(3, 1, 5, 2);
Collections.sort(list); // 기본 오름차순
4.2 내림차순 정렬 (람다식 사용)
list.sort((a, b) -> b - a); // 또는 Collections.sort(list, (a, b) -> b - a);
4.3 절댓값 기준 정렬
list.sort((a, b) -> Math.abs(a) - Math.abs(b));
5. 객체 정렬
5.1 방법 1: Comparable 인터페이스 (자기 자신 기준 정렬)
예시 클래스
class Person implements Comparable<Person> {
String name;
int age;
public Person(String name, int age) {
this.name = name;
this.age = age;
}
@Override
public int compareTo(Person other) {
return this.age - other.age; // 나이 오름차순 정렬
}
}
사용 방법
List<Person> list = new ArrayList<>();
list.add(new Person("Alice", 25));
list.add(new Person("Bob", 20));
Collections.sort(list); // compareTo() 기준으로 정렬됨
5.2 방법 2: Comparator 인터페이스 (외부 기준 정렬)
예시 1: 이름 기준 정렬 (오름차순)
list.sort(Comparator.comparing(p -> p.name));
예시 2: 나이 내림차순 정렬
list.sort((p1, p2) -> p2.age - p1.age);
6. 정리 요약
구분 | 인터페이스 | 사용 위치 | 정렬 기준 변경 |
---|---|---|---|
Comparable |
| 클래스 내부 정의 | 고정 (자기 기준) |
Comparator |
| 외부에서 전달 | 유동적 (다양한 기준 가능) |
Comparable
: "나는 정렬 방법을 내가 정한다!" (클래스 안에서 정의)Comparator
: "정렬 기준은 밖에서 주입한다!" (상황에 따라 유연하게)