[N번째 큰 수 - 백준] / priority_queue , 자료구

Problem

---

https://www.acmicpc.net/problem/2075

2075번: N번째 큰 수

Solve

---

- 원소 입력 시 자동으로 정렬되는 priority_queue 사용

- priority_queue<int, vector<int> , greater<int>> pq;  // 내림차순 정렬되는 pq 선언

- 입력되는 숫자를 삽입

- 조건은 pq의 size가 N보다 작으면 무조건 삽입

- N보다 크면 삽입 후 pq.pop() (가장 작은 수 pop)

- 마지막에는 pq의 size가 N이됨 , 즉 pq.top()이 N번째 큰 수

 

#include <iostream>
#include <queue>

using namespace std;

class sol
{
	int N;
	priority_queue<int, vector<int>, greater<int>> pq;

public:
	sol(const int& n) : N(n)
	{
	}

	void solution()
	{
		int num;
		for (int i = 0; i < N; i++)
		{
			for (int j = 0; j < N; j++)
			{
				cin >> num;
				if (pq.size() < N)
				{
					pq.push(num);
				}
				else
				{
					pq.push(num);
					pq.pop();
				}
			}
		}

		cout << pq.top();
	}
};

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);
	int n;
	cin >> n;
	sol S(n);

	S.solution();
	return 0;
}