【优先队列】POJ1442-Black Box

【思路】

建立一个小堆和一个大堆。大堆用来存放第1..index-1大的数，其余数存放在大堆，小堆的堆顶元素便是我们要求出的第index大的数。每次插入一个A(n)，必须保证大堆中数字数目不变，故先插入小堆中。若此时小堆堆顶小于大堆堆顶，则交换堆顶元素；每次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;//index..最后 
16     priority_queue< int,vector<int>,greater<int> > minHeap;//1..index-1
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 }

 

转载于:https://www.cnblogs.com/iiyiyi/p/4650441.html