kth 概述

Posted 2022-03-26 thefist11

kth既是寻找未排序数组中第k大的数

1. 排序法

将数组排序后，返回第k个元素。

. 复杂度
时间 O(NlogN)
空间 O(1)

2. 优先队列

遍历数组时将数字加入优先队列（堆），一旦堆的大小大于k就将堆顶元素去除，确保堆的大小为k。遍历完后堆顶就是返回值。

. 复杂度
时间 O(NlogK)
空间 O(K)

int findKthLargest(int a[], int nSize, int k)
 
    // Creating a min-heap using priority queue
    priority_queue<int, vector<int>, greater<int> > q;

    for (int i = 0; i < nSize; i++) 
        q.push(a[i]);
        if (q.size() > k)
            q.pop();
    
    return q.top();


void test() 
    int a[] =  1, 4, 5, 2, 10, 8 ;
    int aSize = sizeof(a) / sizeof(int); 
    int k = findKthLargest(a, aSize, 3);
    std::cout << "end" << std::endl;

【引用】

[1] 代码kth.h