Java 정렬 기준과 PriorityQueue

 

Java에서 각종 정렬을 지원하는 Collection은 정렬 순서를 커스터마이징 할 수 있습니다.

기본값은 오름차순 정렬입니다. PriorityQueue는 우선순위큐로써 큐에 특정 값을 넣으면 지정한 로직에 따라 정렬을 해줍니다. 기본값은 꺼낼 때(.poll) 작은 숫자부터 나오게 됩니다.

public static void main(String[] args) {
    Queue<Integer> pq = new PriorityQueue<>();
    pq.add(10);
    pq.add(9);
    pq.add(3);
    pq.add(12);
    while(!pq.isEmpty()) System.out.println(pq.poll());
}

결과

3
9
10
12

 

위 처럼 생략하면 기본값으로 들어갑니다. 기본값은 앞에 숫자에서 뒤에 숫자를 빼서 음수가 나오는지 양수가 나오는지 여부를 가지고 정렬 합니다.

 

Queue<Integer> pq = new PriorityQueue<>((a, b)-> a - b);

 

아래와 같이 a - b 한 결과를 가지고 정렬하게 되어 있습니다.

public static void main(String[] args) {
    Queue<Integer> pq = new PriorityQueue<>((a, b)-> a - b);

    pq.add(10);
    pq.add(9);
    pq.add(3);
    pq.add(12);

    while(!pq.isEmpty()) System.out.println(pq.poll());
}

결과

3
9
10
12

 

하지만 역순(내림차순)으로 정렬 하려면 아래와 같이 바꿔줍니다.

a - b대신 b - a를 해주면 내림차순으로 정렬 됩니다.

public static void main(String[] args) {
    Queue<Integer> pq = new PriorityQueue<>((a, b)-> b - a);
    pq.add(10);
    pq.add(9);
    pq.add(3);
    pq.add(12);
    while(!pq.isEmpty()) System.out.println(pq.poll());
}

결과

12
10
9
3

 

특정 조건으로 정렬하기

아래 코드는 {0, 3}, {1, 9}, {2, 6} 를 arr[0]기준으로 정렬하는 로직입니다.

(a, b) -> a[0] - b[0] 을 기준으로 했기 때문에 arr[0] 기준 오름차순 정렬이 되었습니다.

public static void main(String[] args) {
    int[][] arr = {{0, 3}, {1, 9}, {2, 6}};
    Queue<int[]> pq = new PriorityQueue<>((a, b)-> a[0] - b[0]);
    for (var arr1 : arr) {
        pq.add(arr1);
    }

    while(!pq.isEmpty()) System.out.println(Arrays.toString(pq.poll()));
}

결과

[0, 3]
[1, 9]
[2, 6]

 

내림차순으로 바꾸기

위 코드에서 정렬하는 기준식만 b[0] - a[0]로 바꾸면 arr[0]기준으로 내림차순 정렬 됩니다.

Queue<int[]> pq = new PriorityQueue<>((a, b)-> b[0] - a[0]);

결과

[2, 6]
[1, 9]
[0, 3]