코딩테스트

[leetcode] 215. Kth Largest Element in an Array

nayoon 2024. 5. 28. 00:05

링크

/**
    sort가 없는 상황에서는 heap, priority queue를 쓰면.. 이러한 상황을 극복할 수 있다..
 */

import java.util.PriorityQueue;
import java.util.Collections;

class Solution {
    public int findKthLargest(int[] nums, int k) {
        PriorityQueue<Integer> pq = new PriorityQueue<>(Collections.reverseOrder());

        for(int n: nums) {
            pq.add(n);
        }

        for(int i = 0; i < k - 1; i++) {
            pq.poll();
        }

        return pq.poll();
    }
}