algorithm

2075 - N번째 큰 수

fungod 2022. 1. 30. 12:20
  • priority queue <int,vector<int>,greater<int>>
  • pq의 크기를 N으로 고정
  • 크기가 N보다 커지면 pop
  • 또는 c++에서 algorithm 헤더 파일의 nth_element(fist, nth,last) 를 이용할 수 있다.
  • #include <iostream>
    #include <queue>
    using namespace std;
    priority_queue <int,vector<int>,greater<int>> pq;	
    int N;
    int main() {
        ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    	cin>>N;
    	int count=N,n;
    	for(int i=0;i<N;i++){                 // or for(int i=0;i<N*N;i++)
    		for(int j=0;j<N;j++){
    			cin>>n;
    			pq.push(n);
    			if(pq.size()>N) pq.pop();    
    		}
    	}
    	cout<<pq.top();
    	return 0;
    }