标签:否则 优先队列 判断 ++ amp data- 队列 return 插入
题意
两种操作,第一种可以向序列里加数,第二种查询长度为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
标签:否则 优先队列 判断 ++ amp data- 队列 return 插入
原文地址:https://www.cnblogs.com/jasony/p/12977761.html