1101. Quick Sort (25)

最新推荐文章于 2019-02-07 12:30:20 发布

原创最新推荐文章于 2019-02-07 12:30:20 发布 · 916 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#大根堆 #小根堆 #堆排 #哈希 #hash

PAT 专栏收录该内容

105 篇文章

订阅专栏

本文介绍了一种使用小根堆和大根堆解决pivot元素问题的方法，通过维护两个堆来确保数组左侧构成大根堆，右侧构成小根堆，并在遍历过程中更新堆以满足条件。

1.这道题目需要考虑采用适当的数据结构，即小根堆和大根堆

2.判断某个元素是否能够成为pivot，那么该元素左边数组应该构成一个大根堆，堆顶元素应该小于该元素，该元素的右边构成小根堆，堆顶元素大于该元素

3.左边的小根堆，一直插入即可，利用priority_queue，而右边则需要进行维护，需要编写仿函数

4右边的大根堆维护机制：建立哈希表times，记录每个元素出现的次数，每往后检测新元素数，把新元素出现的次数-1，同时检测大根堆的堆顶元素，如果出现次数为0则弹出，直至剩下出现次数不为0的堆顶

AC代码如下：

//#include<string>
//#include <iomanip>
#include<vector>
#include <algorithm>
//#include<stack>
#include<set>
#include<queue>
#include<map>
//#include<unordered_set>
#include<unordered_map>
//#include <sstream>
//#include "func.h"
//#include <list>
#include<stdio.h>
#include<iostream>
#include<string>
#include<memory.h>
#include<limits.h>
using namespace std;
struct cmp
{
	bool operator()(const int&a, const int&b)
	{
		return a > b;
	}
};
int main(void)
{
	
	int sum;
	cin >> sum;
	int *num = new int[sum];
	map<int, int> times;
	/*vector<int> times(100001, 0);*/
	priority_queue<int> lq;
	priority_queue<int,vector<int>,cmp> rq;
	for (int i = 0; i < sum; i++)
	{
		scanf("%d", &num[i]);
		times[num[i]]++;
		rq.push(num[i]);
	}
	vector<int> ans(0);
	for (int i = 0; i < sum; i++)
	{
		if (i>0) lq.push(num[i - 1]);
		if (rq.size() != 0)
		{//如果右边heap不为空
			times[num[i]]--;//减少num[i]的次数，以维护右边的小根堆
			while (rq.size() != 0 && times[rq.top()] == 0) rq.pop();//检测堆顶元素，出现次数是否为0，如果为0，证明应该被弹出，
			if (rq.size() != 0 && num[i]>rq.top())
			{
				continue;
			}
		}
		if (lq.size() != 0 && num[i] < lq.top())
		{
			continue;
		}
		ans.push_back(num[i]);
	}
	cout << ans.size() << endl;
	sort(ans.begin(), ans.end());
	for (int i = 0; i < ans.size(); i++)
	{
		printf("%d", ans[i]);
		if (i != ans.size() - 1)
			cout << " ";
	}
	cout << endl;//注意在后面添加换行
	return 0;
}