C++最小堆/最大堆的构造

C++最小堆/最大堆的构造

// 使用 priority_queue， #include <queue>
// priority_queue 优先队列，底层数据结构就是红黑树。默认是降序排列，所以本身就是一个最大堆。

#include <iostream>
#include <queue>

using namespace std;

struct node
{
	string name;
	int value;
	node(string name_, int value_) : name(name_), value(value_) {}
	friend bool operator < (const node& a, const node& b)
	{
		return a.value < b.value;  // 定义排序
	} 
};
int main(int argc, char* argv[])
{
	priority_queue<node> p_qu{};  // 
	/*
	do something
	*/
	return 0;
}

// 或者
#include <iostream>
#include <queue>
#include <vector>

using namespace std;

struct node
{
	string name;
	int value;
	node(string name_, int value_) : name(name_), value(value_) {}
};
struct compare
{
	bool operator()(const node& a, const node& b)
	{
		return a.value < b.value;
	}
};
int main(int argc, char* argv[])
{
	priority_queue<node, vector<node>, compare> p_qu{};
	return 0;
}

// 使用 multiset
#include <iostream>
#include <set>
#include <vector>

using namespace std;

struct node
{
	int index;
	int value;
	node(int i, int j) : index(i), value(j) {}
};

struct compare
{
	bool operator()(const node& a, const node& b)  // 参数一定是const
	{
		return a.value < b.value;  // 升序
		// return a.value > b.value;  // 降序
	}
};

int main()
{
	vector<int> nums{2, 8, 4, 1, 9, 5};
	// 最小堆
	multiset<node, compare> m_set{};  // set和multiset底层都是红黑树。compare是自定义的排序（自定义为升序），所以m_set是一个最小堆。
	for (int i = 0; i < nums.size(); i++)
	{
		node n(i, nums[i]);
		m_set.insert(n);
	}
	for (auto it = m_set.begin(); it != m_set.end(); it++)
		cout << it->value << endl;
	return 0;
}

// set会自动去重，multiset允许重复数据。