[백준]K번째 수

백준저지

[K번째 수]https://www.acmicpc.net/problem/11004

 

• C++의 algorithm STL에 있는 nth_element()함수를 이용해서 문제를 풀었다.

• nth_element()는 k번째 수까지 정렬을 하는 좋은 함수이다.

• 시간도 메모리도 상대적으로 많이 썼다.

• 시간이 되면 nth_element함수를 구현해 볼 생각이다.

#include <cstdio>
#include <algorithm>
#include <vector>
using namespace std;

int main(void) {
	int N, K;
	scanf("%d%d", &N, &K);
	vector<long long int> vecArr(N, 0);

	for (int i = 0; i < N; i++) {
		long long int in; scanf("%lld", &in);
		vecArr[i] = in;
	}
	
	nth_element(vecArr.begin(), vecArr.begin() + (K-1), vecArr.end());
	printf("%lld", vecArr[K - 1]);

	return 0;
}