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
/**
题意:
为栈赋予新功能,取第k个值;本题要求k=(n+1)/2
题解:
思路一:巧用STL
用stack判断操作是否合法
使用upper_bound(),lower_bound()函数快速找到需要做改动的值
然后使用vector.erase(),vector.insert()函数进行插入删除。
思路二:线段树
维护这样一棵树:每个节点存储元素个数,找第k个元素时,
如果当前节点lchild.cnt >= k,则说明在左儿子中,递归寻找左儿子
反之 说明在右儿子的第k-lchild.cnt个
*/
///******************************
///**** STL *****
///******************************
#include<bits/stdc++.h>
using namespace std;
const int INF=0x3f3f3f3f;
stack<int>s;
int main(){
vector <int> v;
stack <int> s;
vector <int> :: iterator it;
int n,t;
char a[20];
scanf("%d",&n);
for(int i = 0;i<n;++i){
scanf("%s",a);
if(!strcmp(a,"Pop")){
if(s.empty()) puts("Invalid");
else {
int x = s.top();
s.pop();
printf("%d\n",x);
v.erase(lower_bound(v.begin(),v.end(),x));
}
}else if(!strcmp(a,"Push")){
scanf("%d",&t);
s.push(t);
///注意insert函数是插在当前位置的前面
v.insert(upper_bound(v.begin(),v.end(),t),t);
}else{
if(s.empty())puts("Invalid");
else{
int pos = (s.size()+1)/2-1;
printf("%d\n",v[pos]);
}
}
}
return 0;
}
///******************************
///**** 线段树 *****
///******************************
#include<bits/stdc++.h>
using namespace std;
const int INF=0x3f3f3f3f;
const int maxn = 100000+10;
stack<int>s;
struct node{
int l,r,cnt;
}tree[maxn*4];
void build(int node, int l, int r){
tree[node].l = l;
tree[node].r = r;
tree[node].cnt = 0;
if(l==r) return;
int mid = (l+r)/2;
build(node*2,l,mid);
build(node*2+1,mid+1,r);
}
void update(int node, int ql, int qr,int val){
if(tree[node].l>=ql && tree[node].r<=qr){
tree[node].cnt+=val;
return;
}
int mid = (tree[node].l+tree[node].r)/2;
if(ql<=mid) update(node*2,ql,qr,val);
if(qr>mid) update(node*2+1,ql,qr,val);
tree[node].cnt = tree[node*2].cnt + tree[node*2+1].cnt;
}
int query(int node, int q){
if(tree[node].l == tree[node].r) return tree[node].l;
if(tree[node*2].cnt>=q) query(node*2,q);
else query(node*2+1,q-tree[node*2].cnt);
}
int main(){
int n,t;
char a[20];
build(1,1,maxn);
scanf("%d",&n);
for(int i = 0;i<n;++i){
scanf("%s",a);
if(!strcmp(a,"Pop")){
if(s.empty()) puts("Invalid");
else {
int x = s.top();
s.pop();
printf("%d\n",x);
update(1,x,x,-1);
}
}else if(!strcmp(a,"Push")){
scanf("%d",&t);
s.push(t);
update(1,t,t,1);
}else{
if(s.empty())puts("Invalid");
else{
int pos = (s.size()+1)/2;
printf("%d\n",query(1,pos));
}
}
}
return 0;
}