数组中第K大数字问题

原创于 2014-09-07 17:28:31 发布 · 396 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #排序

面试准备专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一个基于快速排序的分区(partition)思想实现的快速选择算法，该算法用于寻找数组中第K小的元素。通过递归地将数组分为两部分，并根据目标元素的位置决定继续划分哪一部分，最终找到所需元素。文中提供了完整的C++代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

<span style="font-size:24px;color:#FFFF00;">partition  + select</span>


 #include <iostream>

using namespace std;

int Partition(int data[],int start, int end) {
    int i = start;
    int j = end +1;
    int key = data[start];
    while( i < j) {
        i++;
        while( data[i] < key) i++;

        j--;
        while( data[j] > key) j--;

        if( i < j) {
            swap(data[i] , data[j]);
        }
    }
    swap(data[start],data[j]);
    return j;

}

int KthNumber(int data[], int length, int kth) {
    int index = Partition(data,0,length-1);

    while( index != kth) {

        if( index > kth) {
            index = Partition(data,0,index-1);
        } else if( index < kth) {
            index = Partition(data,index +1, length-1);
        }
    }
    return data[index];

}
int main() {

    int length = 10;
    int data[] = {3,45,34,6,4,1,23,9,7,8};
    int kth = 5;
    KthNumber(data,10,kth);


    for( int i =  0 ; i < kth; i++) {
        cout <<data[i] <<"\t";
    }
    cout <<endl;

}