sicily--1775. Simple Sort

双优先级队列排序算法

最新推荐文章于 2022-08-13 10:38:59 发布

原创最新推荐文章于 2022-08-13 10:38:59 发布 · 1.6k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#struct #n2

Sicily 专栏收录该内容

67 篇文章

订阅专栏

本文介绍了一种使用两个优先级队列进行排序的方法。首先通过一个队列存储并排序元素的优先级，然后利用另一个队列根据第一个队列中确定的顺序对实际元素进行排序。文章详细展示了排序过程及C++实现代码。

1.由于最近一直想学号优先级队列，所以还是用了优先级队列

2.由一个优先级对列 seq 来储存最后输入的“优先级”信息，并按“优先级”由小到大排好序；再创建一个优先级队列Q 按照 seq 排好的顺序的“下标”去“排序”元素

3.重点就是两个优先级队列的比较函数 cmp

#include<iostream>
#include<queue>
#include<vector>
#define M 10 //最多的列数
using namespace std;

struct node
{
	int index;
	int priority;
};

struct cmp1
{
	bool operator()(node &n1, node &n2)
	{
		return n1.priority > n2.priority;//"小"的优先
	}
};
priority_queue<node, vector<node>, cmp1> seq;

struct entry
{
	int index;
	int column[M];
};

struct cmp2
{
	bool operator()(entry &e1, entry &e2)
	{
		priority_queue<node, vector<node>, cmp1> original = seq;
		while(!original.empty())
		{
			if(e1.column[original.top().index] != e2.column[original.top().index])
			{
				return e1.column[original.top().index] > e2.column[original.top().index];//小的在前
			}
			else
			{
				original.pop();
			}
		}
		//两个元素完全一致
		return e1.index > e2.index;//小的在前
	}
};

int main()
{
	int testNum;
	cin >> testNum;
	while(testNum--)
	{
		int entryNum, columnNum;
		cin >> entryNum >> columnNum;
		vector<entry> v;
		for(int i = 0; i < entryNum; i++)
		{
			entry temp;
			for(int j = 0; j < columnNum; j++)
			{
				cin >> temp.column[j];
			}
			temp.index = i;
			v.push_back(temp);
		}

		int queryNum;
		cin >> queryNum;
		for(int i = 0; i < queryNum; i++)
		{
			while(!seq.empty())
			{
				seq.pop();//清空
			}
			for(int j = 0; j < columnNum; j++)
			{
				node temp;
				cin >> temp.priority;
				temp.index = j;
				seq.push(temp);
			}
			priority_queue<entry, vector<entry>, cmp2>	Q(v.begin(), v.end());
			while(!Q.empty())
			{
				cout << Q.top().index;
				if(Q.size() == 1)
					;//最后一个元素后面不跟空格
				else
					cout << " ";
				Q.pop();
			}
			cout << endl;
		}
	}
}