pta--特殊堆栈

堆栈是一种经典的后进先出的线性结构，相关的操作主要有“入栈”（在堆栈顶插入一个元素）和“出栈”（将栈顶元素返回并从堆栈中删除）。本题要求你实现另一个附加的操作：“取中值”——即返回所有堆栈中元素键值的中值。给定 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

c++代码

#include<iostream>
#include<string>
#include<stdio.h>
#include<algorithm>
#include<vector>
using namespace std;
int main()
{
    int N;
    scanf("%d",&N);
    string s="";
    int *stc=new int [100010];
    int i=0;
    getline(cin,s);
    vector<int> v;
    while(N--){
        getline(cin,s);
        switch(s[1]){
            case 'o':{
                if(i!=0){
                    printf("%d\n",stc[i-1]);
                    vector<int>::iterator it=lower_bound(v.begin(),v.end(),stc[i-1]);
                    v.erase(it);
                    i--;
                }else{
                    printf("Invalid\n");
                }
            }break;
            case 'e':{
                if(i!=0){
                    printf("%d\n",v[(i+1)/2-1]);
                }else{
                    printf("Invalid\n");
                }
            }break;
            case 'u':{
                int temp=atoi((s.substr(s.find(' '),s.length()-s.find(' '))).c_str());
                stc[i]=temp;
                vector<int>::iterator it =lower_bound(v.begin(),v.end(),temp);
                v.insert(it,temp);
                i++;    
            }
            break;
        }
        s="";
    }
    delete []stc;
    return 0;
}

有两点需要注意的地方

1. 运行超时：这道题我感觉关于取中值这点题意说的不是很清楚，我一开始一直理解为就是中间的值而不是这一组数中第k小的值，所以刚开始一直AC不了。于是之后我就又加了一个临时数组 ，将堆栈数组复制过来后用sort进行排序，但是运行超时了；于是我放弃了sort，加了一个vector数组，在输入数据的时候就进行排序，用到了lower_bound函数，目的是为了找到中值的位置以及插入或者要删除的位置
2. int temp=atoi((s.substr(s.find(' '),s.length()-s.find(' '))).c_str());这句话其实是为了将string转化为int型的数据存入数组，但是其实可以不用这样，可以在读入字符串之后直接将这个数字读进来（当时没想起来…）。