优先队列(堆)经典例题——poj1442 black fox

Posted jasony

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了优先队列(堆)经典例题——poj1442 black fox相关的知识,希望对你有一定的参考价值。

题意

 两种操作,第一种可以向序列里加数,第二种查询长度为k时的序列的第i大元素。

思路   

 优先队列维护最大堆和最小堆分别存放前i-1大的元素前k-i小的元素。

 将当前序列的元素压入最小堆,如果最小堆的最小数大于最大堆的最大数则进行交换,保证最大堆中的所有数小于最小堆。

 因为i值每进行一次自增1,所以每次GET操作后将最小堆中的最小数弹出存入最大堆。

易错点

 Get()的While循环必须放在插入之后进行判断,否则若放在插入之前写作 while (j<N && u[j]==i),当不再插入A(n)时,剩下的Get()将不再执行,导致WA。

代码

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<queue>
 4 using namespace std;
 5 const int MAXN=30000;
 6 int A[MAXN];
 7 int u[MAXN];
 8 int M,N;
 9  
10 int main()
11 {
12     scanf("%d%d",&M,&N);
13     for (int i=0;i<M;i++) scanf("%d",&A[i]);
14     for (int i=0;i<N;i++) scanf("%d",&u[i]);
15     priority_queue< int,vector<int>,less<int> > maxHeap;//大根堆1--(i-1) 
16     priority_queue< int,vector<int>,greater<int> > minHeap;//小根堆i--k 
17     int j=0;
18     for (int i=0;i<M;i++)
19     {
20         minHeap.push(A[i]);//向小根堆插入 
21         if (!maxHeap.empty() && maxHeap.top()>minHeap.top())//大根堆不为空且大根堆堆顶大于小根堆堆顶时,交换 
22         {
23             int temp1=maxHeap.top();maxHeap.pop();
24             int temp2=minHeap.top();minHeap.pop();
25             maxHeap.push(temp2);
26             minHeap.push(temp1);
27         }
28         while (j<N && u[j]==i+1)//此时长度刚好 
29         {
30             cout<<minHeap.top()<<endl;
31             maxHeap.push(minHeap.top());
32             minHeap.pop();
33             j++;
34         }
35     }
36     return 0;
37 }

 

以上是关于优先队列(堆)经典例题——poj1442 black fox的主要内容,如果未能解决你的问题,请参考以下文章

POJ 1442 Black Box(优先队列)

POJ 1442 优先队列 模板

POJ 1442(treap || 优先队列)

动态数组第k小,Poj(1442)

优先队列优化SPFA!!!Layout POJ - 3169挑战例题kuangbin例题

优先队列优化SPFA!!!Layout POJ - 3169挑战例题kuangbin例题