[210413][백준/BOJ] 11004번 K번째 수

문제

입출력


풀이

STL sort를 이용해서 해결할 수 있지만 nth_element를 이용해서도 문제를 해결할 수 있다.
배열은 항상 0부터 시작하니 k가 아닌 k-1로 접근하는것만 신경쓰면 쉽게 해결 가능하다.

코드

#include <bits/stdc++.h>
using namespace std;

int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);

	int n, k, num;
	vector<int> V;
	cin >> n >> k;
	
	while(n--)
	{
		cin >> num;
		V.push_back(num);
	}
	
	nth_element(V.begin(), V.begin() + k - 1, V.end());

	cout << V[k - 1];
}

좋은 웹페이지 즐겨찾기