PAT 1057 Stack

最新推荐文章于 2021-01-03 23:06:56 发布

枫流仁武

最新推荐文章于 2021-01-03 23:06:56 发布

阅读量145

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/weixin_42379312/article/details/94761316

本文介绍了一种使用两个有序集合（一个堆和两个集合）来动态维护一系列整数中位数的算法。该算法在插入新元素时，根据元素大小将其放入不同集合，并在删除元素时调整集合平衡，确保能快速找到当前序列的中位数。

这道题利用两个有序集合,第一个保存大的一半元素,从小到大排列

第二个集合保存小的一半元素,从大到小排列

调了一个小时,终于通过了

#include <iostream>
#include <set>
#include <stack>
#include <vector>
using namespace std;
int main() {
    int N;
    cin>>N;
    stack<int> stack1;
    multiset<int> set1;//由小到大,保存大的元素
    multiset<int,greater<>> set2;//保存小的元素
    vector<string> res;
    for (int i = 0; i < N; ++i) {
        string string1;
        cin>>string1;
        if (string1[1]=='o'){
            if (stack1.empty()){
                res.emplace_back("Invalid");
            } else{
                int val=stack1.top();
                stack1.pop();
                res.push_back(to_string(val));
                if(set1.find(val)==set1.end()){
                    if (set1.size()==set2.size()) {
                        set2.erase(set2.find(val));
                    } else{
                        if(!set2.empty()) {
                            set2.erase(set2.find(val));
                        }
                        int x = *set1.begin();
                        set2.insert(x);
                        set1.erase(set1.find(x));
                    }
                } else{//在set1中
                    if(set1.size()==set2.size()){
                        set1.erase(set1.find(val));
                        int x=*set2.begin();
                        set1.insert(x);
                        set2.erase(set2.find(x));
                    } else{
//                        if (!set2.empty()) {
//                            set2.erase(set2.find(val));
//                        }
//                        int x=*set1.begin();
                        set1.erase(set1.find(val));
//                        set2.insert(x);
                    }
                }
            }
        } else if(string1[1]=='u'){
            int val;
            cin>>val;
            stack1.push(val);
            if (set1.size()==set2.size()){//两个集合当中元素数量相等
                if (set2.empty()||val>=*set2.begin()){
                    set1.insert(val);
                } else{
                    set2.insert(val);
                    int x=*set2.begin();
                    set1.insert(x);
                    set2.erase(set2.find(x));
                }
            } else{
                if (set2.empty()||val>=*set2.begin()){
                    set1.insert(val);
                    int x=*set1.begin();
                    set1.erase(set1.find(x));
                    set2.insert(x);
                } else{
                    set2.insert(val);
                }
            }
        } else{
            if (stack1.empty()){
                res.emplace_back("Invalid");
                continue;
            }
            if (set2.size()==set1.size()){
                res.push_back(to_string(*set2.begin()));
            } else{
                res.push_back(to_string(*set1.begin()));
            }
        }
    }
    for(const auto& x:res){
        cout<<x<<endl;
    }
    return 0;
}