ACwing-786-第k个数-快排思想

最新推荐文章于 2025-10-30 16:47:39 发布

原创最新推荐文章于 2025-10-30 16:47:39 发布 · 127 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法

leetcode 专栏收录该内容

28 篇文章

订阅专栏

本文探讨了如何利用快速排序的思想高效地定位数组中第k个元素的方法。通过递归调用，每次划分确定元素位置，简化查找过程。适合理解排序算法在实际问题中的应用。

在这里插入图片描述

快排思想的运用。因为快排每次可以确定一个元素的最终位置，那么只要确定了第k个数字的位置，递归就可以结束了。

// 第k个数
#include <iostream>
using namespace std;
const int N = 100010;
int n, k, a[N];

int quickFind (int l, int r, int k) {
    // 1. 前半部分和快排完全相同: 找到第k个位置了
    if (l == r ) return a[r];
    // 1.1 以x为枢轴，一次快排的内容
    int i = l - 1, j = r + 1, x = a[l + r >> 1];
    while (i < j) {
        // do i++; while (a[i] < x);
        while (a[++i] < x);
        //do j--; while (a[j] > x);
        while (a[--j] > x);
        if (i < j) swap(a[i], a[j]);
    }
    // 2. 左半边的长度
    int SL = j - l + 1;
    // 2.1 左半边太长了，去左半边找
    if (SL >= k ) return quickFind(l, j, k);
    // 2.2 左半边太短了，去右边找，此时只要去找第K-SL数
    else return quickFind(j + 1, r, k - SL);
}
int main() {
    // 3. 输入n,k，初始化数组
    cin >> n >> k;
    for (int i = 0; i < n; i++) cin >> a[i];
    // 4.直接打印出第k个数
    cout << quickFind(0, n - 1, k) << endl;
    return 0;
}