L3-002. 堆栈
时间限制
200 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
陈越
大家都知道“堆栈”是一种“先进后出”的线性结构,基本操作有“入栈”(将新元素插入栈顶)和“出栈”(将栈顶元素的值返回并从堆栈中将其删除)。现请你实现一种特殊的堆栈,它多了一种操作叫“查中值”,即返回堆栈中所有元素的中值。对于N个元素,若N是偶数,则中值定义为第N/2个最小元;若N是奇数,则中值定义为第(N+1)/2个最小元。
输入格式:
输入第一行给出正整数N(<= 105)。随后N行,每行给出一个操作指令,为下列3种指令之一:
Push keyPop
PeekMedian
其中Push表示入栈,key是不超过105的正整数;Pop表示出栈;PeekMedian表示查中值。
输出格式:
对每个入栈指令,将key入栈,并不输出任何信息。对每个出栈或查中值的指令,在一行中打印相应的返回结果。若指令非法,就打印“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
数组模拟栈,用线段树输出中值。
#include <iostream>
#include <stdio.h>
#include <math.h>
#include <string.h>
#include <algorithm>
#include <vector>
using namespace std;
#define LL long long
#define lt x<<1
#define rt x<<1|1
const int maxn = 1e5+111;
const int inf = 0x3f3f3f3f;
int sum[maxn<<2];
int num[maxn];
void error()
{
printf("Invalid\n");
}
void add(int x,int l,int r,int p,int flag)
{
if(l==r&&p==r){
sum[x]+=flag;
}else{
int mid = (l+r)>>1;
if(p<=mid) add(lt,l,mid,p,flag);
else add(rt,mid+1,r,p,flag);
sum[x]=sum[lt]+sum[rt];
}
}
int query(int x,int l,int r,int cnt)
{
if(l==r){
return l;
}else{
int mid = (l+r)>>1;
if(cnt<=sum[lt]){
return query(lt,l,mid,cnt);
}else{
return query(rt,mid+1,r,cnt-sum[lt]);
}
}
}
int main()
{
int q;
scanf("%d",&q);
int l=1,r=0;
while(q--){
char s[15];
scanf("%s",s);
if(!strcmp("Pop",s)){
if(r<l) error();
else{
printf("%d\n",num[r]);
add(1,1,1e5,num[r],-1);
r--;
}
}else if(!strcmp("PeekMedian", s)){
if(r<l) error();
else{
printf("%d\n",query(1,1,1e5,(r-l+2)/2));
}
}else {
int x;scanf("%d",&x);
num[++r]=x;
add(1,1,1e5,x,1);
}
}
return 0;
}