注意事项:
快速排序的一个拓展算法,建议先了解快排:快速排序—前后双指针法
题目:
给一个未排序的数列,列表中不包含重复元素,要求找到第k大的元素并输出。
输入格式
第一行两个整数,n,k,用空格隔开,分别表示序列长度和需要找到第k大的元素。
第二行有n个元素,用空格隔开,是序列中的所有元素。
输出格式
单个字符,输出序列中第k大的元素。
数据范围
0 < n < 10000;
输入:
10 6
5 3 2 1 4 7 6 9 8 10
输出:
6
#include <cstring>
#include <cmath>
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 10010;
int x[N];
int n, k;
int quick_select(int a[], int l, int r, int k) {
if (l >= r) {return a[k];}
int mid = a[(l+r)/2], i = l-1, j = r+1;
while (i < j) {
while (a[++i] < mid);
while (a[--j] > mid);
if (i < j) swap(a[i], a[j]);
}
if (j >= k) {return quick_select(a, l, j, k);}
else {return quick_select(a, j+1, r, k);}
}
int main() {
cin >> n >> k;
for (int i = 0; i<n; i++) cin >> x[i];
cout << quick_select(x, 0, 9, k-1) << endl;
}
思路:
和二分查找各有优势
1.二分查找用于在已排序的序列中查找给定值的元素。时间复杂度:O(log n)。
优势:对于已排序的序列,二分查找非常高效,因为每次迭代都会将搜索范围缩小一半。
2.快速选择用于在未排序的序列中查找第k小(或第k大)的元素。
优势:对于未排序的序列,快速选择能够高效地找到第k小(或第k大)的元素,而不需要对整个序列进行排序。
如果有所帮助请给个免费的赞吧~有人看才是支撑我写下去的动力!
ps: 前几天在忙着期中考试和各种赶due,明天不出意外应该会恢复更新的,今天水一道吧哎嘿
声明:
算法思路来源为y总,详细请见https://www.acwing.com/
本文仅用作学习记录和交流