洛谷-P1923 【深基9.例4】求第 k 小的数-普及-

P1923 【深基9.例4】求第 k 小的数

题目描述

输入 $n$（$1\le n<5000000$ 且 $n$ 为奇数）个数字 $a_i$（$1\le a_i<{10}^9$），输出这些数字的第 $k$ 小的数。最小的数是第 $0$ 小。

请尽量不要使用 nth_element 来写本题，因为本题的重点在于练习分治算法。

输入格式

第一行有两个整数，分别表示 $n$ 和 $k$。

第二行有 $n$ 个整数，第 $i$ 个数表示 $a_i$。

输出格式

一个整数，表示第 $k$ 小的数。

输入输出样例 #1

输入 #1

5 1
4 3 2 1 5

输出 #1

2

solution

可以用nth_element()函数，其本质上是快速排序的一部分，时间复杂度接近O(n)

代码

#include "iostream"
#include "math.h"
#include "algorithm"
#include "string.h"
#include "unordered_set"using namespace std;
const int N = 5e6;
int A[N];int main() {int n, k;cin >> n >> k;for (int i = 0; i < n; i++) scanf("%d", &A[i]);nth_element(A, A + k, A+n);cout << A[k];return 0;
}

结果