NCPC 2012 Cookie Selection

题目要求每次输出中间的那个数,如果数据很大肯定扛不住;

所以用两个优先队列来维护;

这样的话中间的那个数反正会在两个队列的任何一个的头部;

时间复杂度肯定比较小;

代码:

 1 #include <cstdio>
 2 #include <queue>
 3 using namespace std;
 4 int l1,l2;
 5 priority_queue<int> q1;
 6 priority_queue<int, vector<int>, greater<int> > q2;
 7 void add(int x)
 8 {
 9     q2.push(x);
10     l2++;
11     if(!q1.empty() && !q2.empty())
12     {
13         while(q1.top()>q2.top())
14         {
15             x=q1.top();
16             q1.pop();
17             q1.push(q2.top());
18             q2.pop();
19             q2.push(x);
20         }
21     }
22     if(q1.size()<q2.size())
23     {
24         x=q2.top();
25         q1.push(x);
26         q2.pop();
27         l1++;
28         l2--;
29     }
30 }
31 int get()
32 {
33     int ret;
34     if(l1>l2)
35     {
36         l1--;
37         ret=q1.top();
38         q1.pop();
39     }
40     else
41     {
42         l2--;
43         ret=q2.top();
44         q2.pop();
45     }
46     return ret;
47 }
48 
49 int main()
50 {
51     char str[15];
52     int x;
53     while(scanf("%s",str)!=EOF)
54     {
55         if(str[0]=='#')
56             printf("%d\n",get());
57         else
58         {
59             sscanf(str,"%d",&x);
60             add(x);
61         }
62     }
63     return 0;
64 }
View Code

 

posted @ 2013-10-26 19:53  Yours1103  阅读(317)  评论(0编辑  收藏  举报