L3-002 特殊堆栈(线段树基本应用)

最新推荐文章于 2022-03-21 15:07:08 发布

Helium_wild

最新推荐文章于 2022-03-21 15:07:08 发布

阅读量324

点赞数

CC 4.0 BY-SA版权

分类专栏： PAT

本文链接：https://blog.youkuaiyun.com/qq_42576687/article/details/88907434

PAT 专栏收录该内容

19 篇文章

订阅专栏

本文介绍了一个关于数据结构和算法的问题，即如何维护一个能够快速获取中位数的数据集合。通过使用堆和线段树等数据结构，可以在进行插入和删除操作的同时保持中位数的实时更新。适用于需要频繁获取中位数的应用场景。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述输入样例：
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

这就是套路题。

#include <cstdio>
#include <stack>
using namespace std;
const int N=100000+5;
const int Max=N-5;
int sum[N<<2],ansdex;
stack<int>st;
void update(int n,int val,int l,int r,int pos)
{
    if(l==r)
    {
        sum[pos]+=val;
        return;
    }
    int mid=(l+r)>>1;
    if(n<=mid)
        update(n,val,l,mid,pos<<1);
    else
        update(n,val,mid+1,r,pos<<1|1);
    sum[pos]=sum[pos<<1]+sum[pos<<1|1];
}
void query(int rank_small,int l,int r,int pos)
{
    if(l==r)
    {
        ansdex=l;
        return;
    }
    int mid=(l+r)>>1;
    if(rank_small<=sum[pos<<1])
        query(rank_small,l,mid,pos<<1);
    else
        query(rank_small-sum[pos<<1],mid+1,r,pos<<1|1);
}
char str[20];
int main()
{
    int m;scanf("%d",&m);
    while(m--)
    {
        scanf("%s",str);
        if(str[1]=='o')
        {
            if(st.empty())
                puts("Invalid");
            else
            {
                printf("%d\n",st.top());
                update(st.top(),-1,1,Max,1);
                st.pop();
            }
        }
        else if(str[1]=='u')
        {
            int x;scanf("%d",&x);
            update(x,1,1,Max,1);
            st.push(x);
        }
        else
        {
            if(st.empty())
                puts("Invalid");
            else
            {
                int big=st.size();
                int rank_small=(big&1)?((big>>1)+1):(big>>1);
                query(rank_small,1,Max,1);
                printf("%d\n",ansdex);
            }
        }
    }
}