[LeetCode] Kth Largest Element

1. 문제 : https://leetcode.com/problems/kth-largest-element-in-a-stream/description/

 

2. 풀이

pq의 길이 제한으로 값을 컨트롤하는 방식을 생각해야하는 문제

 

3. 코드

import java.util.*;

public class KthLargest {

    private final PriorityQueue<Integer> q;
    private final int k;

    public KthLargest(int k, int[] nums) {
        this.k = k;
        q = new PriorityQueue<Integer>();
        for (int a :nums) {
            add(a);
        }
    }

    public int add(int val) {
        q.offer(val);
        if (q.size() > k) {
            q.poll();
        }
        return q.peek();
    }
}