C++小顶堆求Topk

2024-05-14 15:58
文章标签 c++ topk 堆求 小顶

本文主要是介绍C++小顶堆求Topk,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

C++小顶堆求Topk

求数组中的Topk数字,比如【1、4、6、7、2、9、8、3、5、0】的Top4是【6、7、8、9】。
用小顶堆来实现,

  • 首先用前4个元素新建一个大小为4的小顶堆,堆顶始终保存堆中的最小值。数组中的剩余数字是【2、9、8、3、5、0】
  • 然后逐个将剩余数字与堆顶比较,如果大于堆顶,则与堆顶交换,并向下调整堆。
  • 最后堆中保存的就是最大的4个数字。

代码如下,MinHeap.cpp:

#include <iostream> 
#include <vector>using namespace std;class MinHeap {private: int maxsize;                // 堆的大小void filterDown(int begin); // 向下调整堆 vector<int> arr;public:MinHeap(int k);        // 构造函数 void createMinHeap(int arr[]); // 创建堆 void insert(int val);  // 插入元素 int getTop();          // 获取堆顶元素 vector<int> getHeap(); //获取堆中的全部元素
};MinHeap::MinHeap(int k) { maxsize = k;
}/*** 创建小顶堆*/
void MinHeap::createMinHeap(int a[]) {for (int i = 0; i < maxsize; i++) {arr.push_back(a[i]);}for(int i = arr.size() / 2 - 1; i >= 0; i--) {filterDown(i);}
}/*** 插入元素*/
void MinHeap::insert(int val) {if (val > getTop()) {arr[0] = val;filterDown(0);}
}/*** 向下调整*/
void MinHeap::filterDown(int current) {int end = arr.size() - 1;int child = current * 2 + 1; // 当前节点的左孩子int val = arr[current];    // 保存当前节点while (child <= end) {// 选出两个孩子中的较小孩子if (child < end && arr[child + 1] < arr[child])child++;if (val < arr[child]) break;else {arr[current] = arr[child]; //孩子节点覆盖当前节点current = child;child = child * 2 + 1;}}arr[current] = val;
}/*** 获取堆顶元素*/
int MinHeap::getTop() {if (arr.size() != 0)return arr[0];return NULL;
}/** * 获取堆中的全部元素*/
vector<int> MinHeap::getHeap() {vector<int> heap;for(int i = 0; i < arr.size(); i++)heap.push_back(arr[i]);return heap;
}int main() {// Test caseint arr[] = {1,4,6,7,2,9,8,3,5,0};int k = 4;MinHeap heap(4); // 创建一个大小为4的堆 heap.createMinHeap(arr);for(int i = k; i < 10; i++) {heap.insert(arr[i]);}cout << "最大的四个元素是" << endl;vector<int> v = heap.getHeap();for(int i = 0; i < v.size(); i++) {cout << v[i] << endl;}return 0;
}

运行结果是:

这里写图片描述

参考:http://blog.csdn.net/u010429424/article/details/77582627
https://www.tuicool.com/articles/NNfMryj

这里写图片描述

这篇关于C++小顶堆求Topk的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/989190

相关文章

C++ 中的 if-constexpr语法和作用

《C++中的if-constexpr语法和作用》if-constexpr语法是C++17引入的新语法特性,也被称为常量if表达式或静态if(staticif),:本文主要介绍C++中的if-c... 目录1 if-constexpr 语法1.1 基本语法1.2 扩展说明1.2.1 条件表达式1.2.2 fa

C++中::SHCreateDirectoryEx函数使用方法

《C++中::SHCreateDirectoryEx函数使用方法》::SHCreateDirectoryEx用于创建多级目录,类似于mkdir-p命令,本文主要介绍了C++中::SHCreateDir... 目录1. 函数原型与依赖项2. 基本使用示例示例 1:创建单层目录示例 2:创建多级目录3. 关键注

C++从序列容器中删除元素的四种方法

《C++从序列容器中删除元素的四种方法》删除元素的方法在序列容器和关联容器之间是非常不同的,在序列容器中,vector和string是最常用的,但这里也会介绍deque和list以供全面了解,尽管在一... 目录一、简介二、移除给定位置的元素三、移除与某个值相等的元素3.1、序列容器vector、deque

C++常见容器获取头元素的方法大全

《C++常见容器获取头元素的方法大全》在C++编程中,容器是存储和管理数据集合的重要工具,不同的容器提供了不同的接口来访问和操作其中的元素,获取容器的头元素(即第一个元素)是常见的操作之一,本文将详细... 目录一、std::vector二、std::list三、std::deque四、std::forwa

C++字符串提取和分割的多种方法

《C++字符串提取和分割的多种方法》在C++编程中,字符串处理是一个常见的任务,尤其是在需要从字符串中提取特定数据时,本文将详细探讨如何使用C++标准库中的工具来提取和分割字符串,并分析不同方法的适用... 目录1. 字符串提取的基本方法1.1 使用 std::istringstream 和 >> 操作符示

C++原地删除有序数组重复项的N种方法

《C++原地删除有序数组重复项的N种方法》给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度,不要使用额外的数组空间,你必须在原地修改输入数组并在使用O(... 目录一、问题二、问题分析三、算法实现四、问题变体:最多保留两次五、分析和代码实现5.1、问题分析5.

C++ 各种map特点对比分析

《C++各种map特点对比分析》文章比较了C++中不同类型的map(如std::map,std::unordered_map,std::multimap,std::unordered_multima... 目录特点比较C++ 示例代码 ​​​​​​代码解释特点比较1. std::map底层实现:基于红黑

C++中函数模板与类模板的简单使用及区别介绍

《C++中函数模板与类模板的简单使用及区别介绍》这篇文章介绍了C++中的模板机制,包括函数模板和类模板的概念、语法和实际应用,函数模板通过类型参数实现泛型操作,而类模板允许创建可处理多种数据类型的类,... 目录一、函数模板定义语法真实示例二、类模板三、关键区别四、注意事项 ‌在C++中,模板是实现泛型编程

利用Python和C++解析gltf文件的示例详解

《利用Python和C++解析gltf文件的示例详解》gltf,全称是GLTransmissionFormat,是一种开放的3D文件格式,Python和C++是两个非常强大的工具,下面我们就来看看如何... 目录什么是gltf文件选择语言的原因安装必要的库解析gltf文件的步骤1. 读取gltf文件2. 提

C++快速排序超详细讲解

《C++快速排序超详细讲解》快速排序是一种高效的排序算法,通过分治法将数组划分为两部分,递归排序,直到整个数组有序,通过代码解析和示例,详细解释了快速排序的工作原理和实现过程,需要的朋友可以参考下... 目录一、快速排序原理二、快速排序标准代码三、代码解析四、使用while循环的快速排序1.代码代码1.由快