I - 郭大侠与线上游戏
Time Limit: 6000/2000MS (Java/Others) Memory Limit: 125535/65535KB (Java/Others)
曾在网络游戏告白,但对方是假冒女生的人妖,而对此有阴影的郭大侠,坚持把游戏和现实分得清清楚楚。有一天竟被网络游戏中的女玩家告白了,虽然在游戏中接受了亚子的表白而结婚,但仍然不敢确信亚子就是真真正正的女生。终于有一天,卢西安所属四人公会的富豪会长决定举行线下聚会,结果发现四个人居然都是同一所高中的学生,甚至另外三人都是真真正正的可爱女孩子。
郭大侠还没完全接受这个事实时,他们发现亚子居然是重度的现实游戏混淆症患者,不仅在现实生活中用游戏角色名字“卢西安”叫人,还旁若无人黏住英骑身上。为了让亚子能够恢复正常,四人的游戏生活延续至学校。
……
今天,拯救亚子的计划是玩游戏!
这个游戏是这样的~
有一个像队列一样的东西,你可以push一个元素到这个队列的尾部,也可以pop掉这个队列的第一个元素
现在问题来了,这个队列中位数是多少呢?
中位数的定义为该队列升序排序后第k/2+1个的数的值,k为这个队列的大小
Input
第一行一个 n ,表示有n个操作
接下来 n 行为这三个操作之一:
1 x,向这个队列推入一个数x
2,弹出这个队列的第一个数
3,查询这个队列的中位数是多少
1<=n<=1000000 ,保证输入都是在int范围的整数。
保证队列中的每个数都不一样哦~
Output
对于每一个询问输出答案。
保证有解~
Sample input and output
Sample Input | Sample Output |
---|---|
6 1 2 3 1 4 3 2 3 | 2 4 4 |
Source
2016 UESTC Training for Data Structures Problem I
My Solution
用2个set,Lset维护最小的n/2个数,Rset维护其余的数;
然后用queue来记录这些数的数序;
每次维护都保证Lset里面是最小的n/2个数就好了,
然后查询的时候输出*Rset.begin()就好了。
复杂度 O(n)
但是TLE了4发,(┬_┬)
一不小心用了cout ……
其实以前基本上只在调试的时候用cout的-_-||
#include <iostream>
#include <cstdio>
#include <set>
#include <queue>
using namespace std;
set<int> Lset, Rset;
queue<int> q;
int main()
{
#ifdef LOCAL
freopen("a.txt", "r", stdin);
#endif // LOCAL
int n, op, x;
scanf("%d", &n);
while(n--){
scanf("%d", &op);
if(op == 1){
scanf("%d", &x);
q.push(x);
if(Rset.empty()) {Rset.insert(x);}//cout<<*(--Rset.end())<<endl;}
else if(Lset.empty()){
if(x > *Rset.begin()){
Lset.insert(*Rset.begin());
Rset.erase(Rset.begin());
Rset.insert(x);
}
else
Lset.insert(x);
}
else{
if(x >= *Rset.begin()){
if(Rset.size() == Lset.size())
Rset.insert(x);
else{
Rset.insert(x);
Lset.insert(*Rset.begin());
Rset.erase(Rset.begin());
}
}
else{
if(Rset.size() == Lset.size()){
Lset.insert(x);
Rset.insert(*Lset.rbegin());
Lset.erase(--Lset.end());
}
else{
Lset.insert(x);
}
}
}
}
else if(op == 2){
x = q.front();
q.pop();
//cout<<*(--Lset.end())<<endl;
if(Lset.empty()){
Rset.erase(Rset.begin());
}
else{
if(x >= *Rset.begin()){
if(Rset.size() == Lset.size()){
Rset.erase(Rset.lower_bound(x));
Rset.insert(*Lset.rbegin());
Lset.erase(--Lset.end());
}
else{
Rset.erase(Rset.lower_bound(x));
}
}
else{
if(Rset.size() == Lset.size()){
Lset.erase(Lset.lower_bound(x));
}
else{
Lset.erase(Lset.lower_bound(x));
Lset.insert(*Rset.begin());
Rset.erase(Rset.begin());
}
}
}
}
else{
printf("%d\n", *Rset.begin());
}
}
return 0;
}
Thank you!