POJ_1442_Black Box(堆)
1 # include <stdio.h> 2 # include <queue> 3 # include <vector> 4 #include<functional> 5 using namespace std; 6 const int MAX = 30001; 7 int A[MAX]; 8 int main() 9 { 10 int m,n,i,j,k; 11 priority_queue<int> q2; //最大堆(默认); 12 priority_queue<int,vector<int>,greater<int> > q1;//最小堆(自定义); 13 scanf("%d%d",&n,&m); 14 for(i=0;i<n;i++) 15 scanf("%d",&A[i]); 16 j=0; 17 for(i=0;i<m;i++) 18 { 19 scanf("%d",&k); 20 while(j<k) 21 { 22 q1.push(A[j]); 23 if(!q2.empty() && q1.top()<q2.top()) 24 { 25 int t; 26 t=q1.top(); 27 q1.pop(); 28 q1.push(q2.top()); 29 q2.pop(); 30 q2.push(t); 31 } 32 j++; 33 } 34 printf("%d\n",q1.top()); 35 q2.push(q1.top()); 36 q1.pop(); 37 } 38 return 0; 39 }
http://poj.org/problem?id=1442
/**************************************************************************/
/*
题意:
输入一行数A[];再输入一行数u[];让你输出前[]的前u[p]个数据中,第p+1小的数
正如给出的例子
7 4
3 1 -4 2 8 -1000 2
1 2 6 6
前1个数中第1小的数是3;
前2个数中第2小的数是3;
前6个数中第3小的数是1;
前6个数中第4小的数是2;
怎么来做呢用的就是维护好两个堆,堆用优先队列来实现。
一个最大堆,一个最小堆。最大堆意思就是最大的元素在堆顶;最大堆存的是前i-1
小的数,那么最小堆的堆顶不就是要输出的第i小的数吗。
在这里我们用一个循环循环的作用就是每一次往最大堆离放i-1(或者i反正就是那么个
意思)个元素,代表前i-1小的数,那么我们循环里面再套一个while循环,因为你是在
1-u[p]这么多元素里找最小值,这些元素总得输入吧。每输入一个数据先放到q1(小)里
那么放进去之后堆顶元素就是这些数据当中最小的,如果最小的那个数比最大堆的堆顶
元素小,说明前i-1个最小的数据需要更新,只要将小堆堆顶和大堆堆顶元素交换即可。
那么在输出的时候同样,一个for循环结束后,最大堆的前i-1个最小的需要增至i个直接
把最小堆的堆顶元素赋值给最小堆最大堆堆顶弹出,
*/
/***************************************************************************/