1057. Stack (30)

题目连接：https://www.patest.cn/contests/pat-a-practise/1057

题意：在堆栈中增加一个返回中位数的功能。

一开始我就直接将堆栈中的数移到vector中然后进行排序……

网上看了很多人的答案，根本看不懂，树状数组好难理解啊，等以后再细细理解吧。

参考了http://blog.csdn.net/kakitgogogo/article/details/51926600的解法，高明之处在于省去了将堆栈中元素移动的时间，利用set(默认由小到大排序)，在Push的时候进行判断然后分别压入s1,s2(s2大于mid,s1小于等于mid);

然后Pop和Push操作的时候分被调整s1,s2以及mid的值。保证s1的大小等于s2的大小或者等于s2的大小+1，这样，s1中的最后一个值就是mid;

注意删除的时候要删除的对象是指针，否则将会把所有值相同的元素都删去

#include<stdio.h>
#include<set>
#include<string.h>
#include<stack>
using namespace std;
int mid=100005;
multiset<int>s1,s2;

void adjust()
{
    multiset<int>::iterator it;
    if (s1.size()<s2.size())
    {
        it=s2.begin();
        s1.insert(*it);
        s2.erase(it);
    }
    if (s1.size()>s2.size()+1)
    {
        it=s1.end();
        it--;
        s2.insert(*it);
        s1.erase(it);
    }
    if (!s1.empty())
    {
        it=s1.end();
        it--;
        mid=*it;
    }
}

int main()
{
    int N;
    scanf("%d",&N);
    char s[15];
    stack<int>st;
    int tmp,v;
    while (N--)
    {
        scanf("%s",s);
        if (strcmp(s,"Pop")==0)
        {
            if (st.size()==0)printf("Invalid
");
            else{
                tmp=st.top();
            printf("%d
",tmp);
            st.pop();
            if (tmp<=mid)s1.erase(s1.find(tmp));
            else s2.erase(s2.find(tmp));
            adjust();
                }
        }
        else if (strcmp(s,"PeekMedian")==0)
        {
           if (st.size()==0)printf("Invalid
");
           else
           {
               printf("%d
",mid);
               //adjust();
           }
        }
        else if (strcmp(s,"Push")==0)
        {
            scanf("%d",&v);
            st.push(v);
            if (v<=mid)s1.insert(v);
            else s2.insert(v);
            adjust();
        }
    }
    return 0;
}