POJ 1442 Black Box 优先队列

题意:给定M个数,每次可以插入序列一个数;再给N个数,表示在插入第几个数时输出一个数,第一次输出序列中最小的,第二次输出序列中第二小的……以此类推,直到输出N个数。

分析:因为输出时是按照先输出最小的,再输出第二小这样的方式输出的,相当于依次输出一个有序序列中的值。但因为这个序列不是固定不变的,而是不断的在更新,所以用数组是无法实现的。我们可以用优先队列来做。

定义两个优先队列,一个用来存储前k小的数,大数在前,小数 在后;另一个优先队列第k+1小到最大的数,小数在前,大数在后。每次拿到一个数,先判断第一个优先队列中的数满不满k个,如果不满k个,则直接把这个数 压入到第一个队列;如果满k个,判断这个数和第一个优先队列中的第一个数的大小:如果比第一个数大,就压入第二个优先队列;如果比第一个数小,就把第一个 优先队列的队首元素弹出压入第二个队列,把这个新数压入第一个优先队列。

输出时,如果第一个优先队列里的元素个数小于k,则先把第二个优先队列里的队首元素弹出压入第一个优先队列,然后输出第一个优先队列的队首元素;如果满k个,则直接输出第一个优先队列的队首元素。

 

 1 #include<stdio.h>
 2 #include<cstring>
 3 #include<iostream>
 4 #include<map>
 5 #include<vector>
 6 #include<deque>
 7 #include<queue>
 8 using namespace std;
 9 
10 int a[30010],b[30010];
11 int main()
12 {
13     int n, m, i, j, k, x, ans;
14     while(~scanf("%d%d",&m,&n))
15     {
16         priority_queue<int, vector<int>, less<int> > que1;
17         priority_queue<int, vector<int>, greater<int> > que2;
18         for(i = 1; i <= m; i++)
19             scanf("%d",&a[i]);
20         for(j = 1; j <= n; j++)
21             scanf("%d",&b[j]);
22         i = 0;
23         j = k = 1;
24         while(j <= n)
25         {
26             if(i == b[j])  //弹出第k小的数
27             {
28                 j++;
29                 if(que1.size() < k) //que1里的元素不够k个
30                 {
31                     x = que2.top();
32                     que1.push(x);
33                     que2.pop();
34                 }
35                 ans = que1.top();
36                 printf("%d\n",ans);
37                 k++;  //每次弹出一个数后,k的值都要加1
38             }
39             else
40             {
41                 i++;
42                 //que1里的元素不够k个
43                 if(que1.size() < k)
44                 {
45                     que2.push(a[i]);
46                     x = que2.top();
47                     que2.pop();
48                     que1.push(x);  //先把a[i]压入que2,再从que2里取出最小值,压入que1
49                 }
50                 //如果que1的元素达到k个,且要压入队列的值比que1中的当前最大值大,说明que1中当前的最大值并不是第k小
51                 else if(que1.top() > a[i])
52                 {
53                     x = que1.top();
54                     que1.pop();
55                     que2.push(x);
56                     que1.push(a[i]);
57                 }
58                 //que1中的元素个数达到k个,且要压入队列的值比que1中的当前最大值小,说明que1中当前的最大值就是是第k小,则把a[i]直接压入que2中
59                 else
60                 {
61                     que2.push(a[i]);
62                 }
63             }
64         }
65     }
66     return 0;
67 }
View Code

 

 

 

posted @ 2015-10-25 19:21  yyblues  阅读(274)  评论(0编辑  收藏  举报