PriorityQueue

🤔코테 문제를 풀다보면 우선순위큐를 사용할 때가 많다. 그래서 한번 정리할 필요가 있어 글을 쓰게 되었다.

PriorityQueue는 우선순위큐라고도 불리며 내부는 힙으로 구현되어 있어 힙의 성질을 가지고 있다.
PriorityQueue를 사용하려면 아래와 같이 선언하면 된다.

생성

PriorityQueue<Integer> queue = new PriorityQueue<>();

---

offer

😑큐에 값을 넣으려면 add함수를 이용하면 된다.

 PriorityQueue<Integer> queue = new PriorityQueue<>();
 queue.offer(1);
 queue.offer(2);
 queue.offer(3);

---

poll

😎큐에서 값을 빼오려면 poll함수를 이요하면 된다. 이 때 정렬에 따라 가장 큰 값 는 가장 작은 값이 나오게 되는데 여기서 아무 값도 주지 앟는 다면 가장 작은 값이 나온다.

PriorityQueue<Integer> queue = new PriorityQueue<>();
        queue.offer(1);
        queue.offer(2);
        queue.offer(3);
        Integer poll = queue.poll();
        System.out.println("poll = " + poll);
        // 1

😊만약 값을 제거하지 않고 확인만 하고 싶다면 peek을 이용하면된다.

PriorityQueue<Integer> queue = new PriorityQueue<>();
        queue.offer(1);
        queue.offer(2);
        queue.offer(3);
        
        System.out.println("poll = " + queue.peek());
        // 1

---

Comparator구현

👻큐를 사용하다보면 내가 직접 정렬 순서를 구성하고 싶어지는 경우가 있다. 그럴때는 아래와 같은 방법을 이용하면 된다.

PriorityQueue<Integer> queue = new PriorityQueue<>(new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                return o2-o1;
            }
        });
        queue.offer(1);
        queue.offer(2);
        queue.offer(3);
        Integer poll = queue.poll();
        System.out.println("poll = " + poll);
        //3

🤨comparator를 직접구현하여 집어넣어주면 되는데 여기서는 내림차순으로 정렬을 하였다.
위의 경우 람다식으로도 해결 할 수 있다.

 PriorityQueue<Integer> queue = new PriorityQueue<>((o1, o2) -> o2-o1);
        queue.offer(1);
        queue.offer(2);
        queue.offer(3);
        Integer poll = queue.poll();
        System.out.println("poll = " + poll);
        // 3

이 외에도 단순하게 내림차순으로 정렬을 하고 싶다면 Collections.reverseOrder()을 넣어주면 된다.

PriorityQueue<Integer> queue = new PriorityQueue<>(Collections.reverseOrder());

😎queue에서 사용할 수 있는 함수는 사용할 수 있으니 만약 필요하다면 사용하면 된다.