TopK算法

最新推荐文章于 2024-08-13 12:00:43 发布

CoderZhuuu

最新推荐文章于 2024-08-13 12:00:43 发布

阅读量1.4k

点赞数

分类专栏：算法 c++ 文章标签： c++ 算法堆 TopK

本文链接：https://blog.csdn.net/codermaximum/article/details/101382162

版权

本文介绍了如何使用C++通过最小堆来实现TopK算法，具体步骤包括从文件‘testData.txt’中读取M和N的值，然后利用堆数据结构找到前K个最大元素。

摘要由CSDN通过智能技术生成

需要先在 "testData.txt" 文件里面输入M和N

比如2000 10

通过最小堆实现TopK

最小堆通过数组模拟二叉树来实现

#pragma warning(disable:4996)
#pragma once
#include <fstream>
#include <ctime>

const char* FILEIN = "testData.txt";
const char* FILEOUT = "topKResult.txt";
const int BUFFERSIZE = 1024;

template<class T>
void swap(T & a, T & b) {
	T c = a;
	a = b;
	b = c;
}

template<class T>
class MinHeap {
public:
	MinHeap(int n = 0);
	MinHeap(const MinHeap<int> & minHeap);
	~MinHeap();
	bool full() const { return m_capacity == m_size; }
	bool empty() const { return 0 == m_size; }
	const size_t size() const { return m_size; } // 返回当前堆的大小
	const T & top() const { return m_elem[0]; } // 返回最小的元素
	void insertElem(const T & elem);
	void deleteElem();
	void resize(int n);
	void print(std::ofstream & fout);
private:
	T * m_elem;
	size_t m_size;