用multiset/priority_Queue来实现最大最小堆

最新推荐文章于 2021-07-20 09:38:11 发布

benbenab

最新推荐文章于 2021-07-20 09:38:11 发布

阅读量3.8k

点赞数 1

分类专栏： others 基本概念

本文链接：https://blog.csdn.net/benbenab/article/details/8214817

版权

本文讨论如何利用STL中的multiset和priority_queue来实现最大堆和最小堆，特别是在解决KNN问题中的应用。介绍了multiset的常见操作，并提到了priority_queue作为堆的另一种形式，强调了其在数据处理中的效率和便捷性。

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

面试题：实现KNN，给定各个点之间的距离，返回某个点的k个neighbor。

思路：考虑到A家喜欢问海量数据，所以用heap，复杂度O(nlogk)。当时是自己实现的heap。后来发现可以直接调用STL中的set 或者multiset来实现。

Mulitiset: STL中 set 和multiset

set的含义是集合，它是一个有序的容器，里面的元素都是排序好的，支持插入，删除，查找等操作，就 像一个集合一样。所有的操作的都是严格在logn时间之内完成，效率非常高。 set和multiset的区别是：set插入的元素不能相同，但是multiset可以相同。（本例中使用距离作为比较元素，而距离是很有可能）

typedef struct{
	int index;
	double length;
}Info;

typedef struct{
	bool operator()(const Info& info1,  const Info& info2){
		if ((info1.length - info2.length) > 0)
			return true;
		return false;
	}
}Comp;
typedef multiset<Info,Comp> intSet;
void GetLeastNumbers(const vector<Info