(C++) - LeetCode (easy) 703. Kth Largest Element in a Stream

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

Kth Largest Element in a Stream - LeetCode

📕 풀이방법

📔 입력 및 초기화

큰 원소를 pq top으로 우선순위를 가지는 queue pq와 k를 선언 후 class 생성자에서 nums의 원소를 받아 값을 저장해줍니다.

📔 풀이과정

pq의 top은 항상 k번째로 큰 수가 저장되도록 add함수에서 val을 push후 pq.size() > k인 동안 원소에서 pop해줍니다.

📔 정답 출력 | 반환

pq의 top을 반환합니다.

---

📕 Code

📔 C++

class KthLargest {
public:
    int k;
    priority_queue <int, vector<int>, greater<int>> pq;
    
    KthLargest(int k, vector<int>& nums) {
        this->k = k;
        for(auto n : nums) pq.push(n);
    }
    
    int add(int val) {
        pq.push(val);
        while(pq.size() > k) {
            pq.pop();
        }
        return pq.top();
    }
};

/**
 * Your KthLargest object will be instantiated and called as such:
 * KthLargest* obj = new KthLargest(k, nums);
 * int param_1 = obj->add(val);
 */

 

---

*더 나은 내용을 위한 지적, 조언은 언제나 환영합니다.