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
用一个线段树来维护某段区间上所有数的个数,刚开始以为要用到主席树,后来看到数值的取值只有1e5,直接用线段树来维护这个大区间就可以了。详见代码
#include<iostream> #include<bits/stdc++.h> using namespace std; int tree[1123450]; stack<int>A; void update(int left,int right,int rt,int value,int add) { // cout<<rt<<endl; if(left==right) { tree[rt]+=add; return ; } int mid=(left+right)>>1; if(value<=mid) update(left,mid,rt<<1,value,add); else update(mid+1,right,rt<<1|1,value,add); tree[rt]=tree[rt<<1]+tree[rt<<1|1]; } int qu(int left,int right,int rt,int value) { if(left==right) return left; int mid=(left+right)>>1; if(tree[rt<<1]>=value) return qu(left,mid,rt<<1,value); else return qu(mid+1,right,rt<<1|1,value-tree[rt<<1]); } int main() { int n; cin>>n; memset(tree,0,sizeof(tree)); char ch[66]; while(!A.empty()) A.pop(); while(n--) { scanf("%s",&ch); if(!strcmp(ch,"Pop")) { if(A.empty()) printf("Invalid\n"); else { int tt=A.top(); A.pop(); printf("%d\n",tt); update(0,100000,1,tt,-1); } } else if(!strcmp(ch,"PeekMedian")) { if(A.empty()) { printf("Invalid\n"); continue; } int sz=A.size(); if(sz%2) sz=(sz+1)/2; else sz=sz/2; printf("%d\n",qu(0,100000,1,sz)); } else if(!strcmp(ch,"Push")) { int x; scanf("%d",&x); A.push(x); update(0,100000,1,x,1); } } return 0; }