topK问题,取数组中第k大和前k个大的元素--每日一题

分析
用到大顶堆、小顶堆的知识点,正好复习一下:
求第k大和前k个大的元素要用大顶堆还是小顶堆呢?
如果直观的用大顶堆的话,堆顶是堆的最大值,而我们需要选出前K个最大的数据。如果我们在建堆时最大值被放到了对顶,那之后的k-1个次大值无法放入堆中。所以,选用最大堆只能找出k中的一些最大值,而不是全部。
所以我们选用小顶堆,如果大于堆顶最小值则替换堆顶值放入堆中,再调整为最小堆,最终小顶堆存放前k个大元素,堆顶是第k个大元素。

代码

//
//  27-TopK.cpp
//  Cracking-the-code
//
//  Created by mac on 2019/3/28.
//  Copyright © 2019 iris. All rights reserved.
//

#include <iostream>
using namespace std;

void swap(int * a, int i, int j) {
    a[i] = a[i]^a[j];
    a[j] = a[i]^a[j];
    a[i] = a[i]^a[j];
}

//对parent及其子节点调整为最小堆O(logK)
void adjustDown(int *a, int parent, int K) {
    int child = parent * 2 + 1;
    while (child < K) {
        //和左右孩子节点中中最小的那个交换
        if (a[child] > a[child+1]) {
            ++child;
        }
        if (a[parent] > a[child]) {
            swap(a, parent, chil
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值