STL大法之nth_element

STL大法好

刷luogu遇到排序TLE了，看到一个厉害的函数记录一下

题目链接
传送门https://www.luogu.com.cn/problem/P1923

题目描述

神秘代码

函数语句：nth_element(数组名,数组名+第k小元素,数组名+元素个数)

解决代码

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
int n,m,a[5000005];
int main(){
	cin >> n >> m;
	for(int i = 1;i <= n;i++){
		scanf("%d",&a[i]);
		}
	nth_element(a+1,a+1+m,a+1+n);
	cout << a[m+1] << endl;
	return 0;
	}