L3-002 特殊堆栈 (30分)
堆栈是一种经典的后进先出的线性结构,相关的操作主要有“入栈”(在堆栈顶插入一个元素)和“出栈”(将栈顶元素返回并从堆栈中删除)。本题要求你实现另一个附加的操作:“取中值”——即返回所有堆栈中元素键值的中值。给定 N 个元素,如果 N 是偶数,则中值定义为第 N/2 小元;若是奇数,则为第 (N+1)/2 小元。
输入格式:
输入的第一行是正整数N≤10^5。随后 N 行,每行给出一句指令,为以下 3 种之一:
Push key
Pop
PeekMedian
其中 key 是不超过 10
5
的正整数;Push 表示“入栈”;Pop 表示“出栈”;PeekMedian 表示“取中值”。
输出格式:
对每个 Push 操作,将 key 插入堆栈,无需输出;对每个 Pop 或 PeekMedian 操作,在一行中输出相应的返回值。若操作非法,则对应输出 Invalid。
输入样例:
17
Pop
PeekMedian
Push 3
PeekMedian
Push 2
PeekMedian
Push 1
PeekMedian
Pop
Pop
Push 5
Push 4
PeekMedian
Pop
Pop
Pop
Pop
输出样例:
Invalid
Invalid
3
2
2
1
2
4
4
5
3
Invalid
思路其实简单的说1e5的数据没有办法直接暴力的跑暴力这跑必会时间超限,所以我们被迫换个思路,就是用树状数组记录他每个点维护的数值,然后二分查询结果让求得个数居中就行。
/**
_ ___ _______
| |/ (_) |___ / |
| ' / _ _ __ __ _ / /| |__ __ _ _ __ __ _
| < | | '_ \ / _` | / / | '_ \ / _` | '_ \ / _` |
| . \| | | | | (_| |/ /__| | | | (_| | | | | (_| |
|_|\_\_|_| |_|\__, /_____|_| |_|\__,_|_| |_|\__, |
__/ | __/ |
|___/ |___/
**/
#include <stack>
#include <map>
#include <queue>
#include <string>
#include<iostream>
#include<stdio.h>
#include<string.h>
#include <algorithm>
#include <math.h>
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
typedef pair<ll, ll> pii;
#define mem(a,x) memset(a,x,sizeof(a))
#define debug(x) cout << #x << ": " << x << endl;
#define rep(i,n) for(int i=0;i<(n);++i)
#define repi(i,a,b) for(int i=int(a);i<=(b);++i)
#define repr(i,b,a) for(int i=int(b);i>=(a);--i)
const int maxn = 1e5 + 10;
#define inf 0x3f3f3f3f
#define sf scanf
#define pf printf
stack<int> s;
string str;
int sum[maxn]={0};
int lowbit(int x){
return (x & (-x));
}
void add(int x,int y){
for(int i=x;i<maxn;i+=lowbit(i)){
sum[i]+=y;
}
}
int getsum(int x){
int num=0;
for(int i=x;i>=1;i-=lowbit(i)){
num+=sum[i];
}
return num;
}
void ack(){
int l=1,r=maxn,k=(s.size()+1)/2;
while(l<r){
int mid=(l+r)/2;
if(getsum(mid)>=k){
r=mid;
}else l=mid+1;
}
pf("%d\n",r);
}
int main(){
int n;
cin>>n;
while(n--){
cin>>str;
if(str=="Pop"){
if(s.empty()){
puts("Invalid");
}else {
add(s.top(),-1);
pf("%d\n",s.top());
s.pop();
}
}
else if(str=="Push"){
int m;
sf("%d",&m);
s.push(m);
add(m,1);
}else if(str=="PeekMedian"){
if(s.empty()){
puts("Invalid");
}else ack();
}
}
return 0;
}