L3-002 特殊堆栈（30 分)

不念不忘少年蓝@ 2022-03-08 01:06 195阅读 0赞

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

用一个Vector计算中值，插入的时候用二分查找找到相应的位置，保持v有序。

#include<stdio.h>
    #include<string.h>
    #include<math.h>
    #include<iostream>
    #include<string>
    #include<algorithm>
    #include<map>
    #include<set>
    #include<queue>
    #include<stack>
    #include<vector>
    using namespace std;
    #define inf 0x3f3f3f3f
    #define LL long long
    int main()
    {
    	int n,i,j,x;
    	string ss;
    	stack<int> s;
    	vector<int> v;
    	vector<int>::iterator it;
    	cin>>n;
    	while(n--)
    	{
    		cin>>ss;
    		if(ss=="Pop")
    		{
    			if(s.empty())
    			cout<<"Invalid"<<endl;
    			else
    			{
    				cout<<s.top()<<endl;
    				it = lower_bound(v.begin(),v.end(),s.top());
    				v.erase(it);
    				s.pop();
    			}
    		}
    		else if(ss=="PeekMedian")
    		{
    			if(s.empty())
    			cout<<"Invalid"<<endl;
    			else
    			{
    				cout<<v[(s.size()-1)/2]<<endl;
    			}
    		}
    		else
    		{
    			cin>>x;
    			s.push(x);
    			it = lower_bound(v.begin(),v.end(),x);
    			v.insert(it,x);
    		}
    	}
    	
    }