11004_K번째 수

정렬을 하여 k번째에 뭔가 있는지 찾는 문제이다. 간단할 줄 알았으나, N이 5000000이기 때문에 기존 방법으로는 안되었다. 나는 굳이 기존 방법을 하고 시간 초과를 겪었다. sort에서 에러가 나는가 보다. 

 

#include<iostream>
#include<algorithm>
#include<vector>
 
using namespace std;
 
int main(void){
    int n, k;
    vector<int> v;
 
    cin >> n >> k;
    for(int i = 0; i < n; i++){
        int num;
        cin >> num;
        v.push_back(num);
    }
    stable_sort(v.begin(), v.end());
    cout << v[k - 1] << "\n";
}
 

하핫

열심히 고민해보다가 그냥 입출력에서 시간을 많이 잡아먹는 줄 알게 되었다. 

cin.tie(NULL);
ios::sync_with_stdio(false);

을 넣어주니 해결!

#include<iostream>
#include<algorithm>
#include<vector>
 
using namespace std;
vector<int> v;
 
int main(void){
    cin.tie(NULL);
    ios::sync_with_stdio(false);
    int n, k;
 
    cin >> n >> k;
    for(int i = 0; i < n; i++){
        int num;
        cin >> num;
        v.push_back(num);
    }
    sort(v.begin(), v.end());
    cout << v[k - 1] << "\n";
}