poj3481 Double Queue(splay)

最新推荐文章于 2018-05-31 22:54:26 发布

原创最新推荐文章于 2018-05-31 22:54:26 发布 · 370 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#poj

poj 同时被 2 个专栏收录

15 篇文章

订阅专栏

平衡树

10 篇文章

订阅专栏

本文介绍了一种使用平衡树解决双端队列问题的方法。通过平衡树的数据结构，可以高效地完成插入、删除最大值及最小值的操作。文章提供了完整的C++代码实现，并通过POJ3481问题验证了该方法的有效性。

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

题面在这里
题目大意：
有一些人，每个人有一个名字和优先级。初始时队列为空。现在有种操作：插入一个人；弹出优先级最大的那个人；弹出优先级最小的那个人。

做法：
平衡树练习。直接上代码。

/*************************************************************
    Problem: poj 3481 Double Queue
    User: fengyuan
    Language: C++
    Result: Accepted
    Time: 204 ms
    Memory: 1.8 MB
    Submit_Time: 2017-12-09 17:23:38
*************************************************************/

#include<cstdio>
#include<cstring>
#define rep(i, x, y) for (int i = (x); i <= (y); i ++)
#define down(i, x, y) for (int i = (x); i >= (y); i --)
#define mid ((l+r)/2)
#define lc (o<<1)
#define rc (o<<1|1)
#define pb push_back
#define mp make_pair
#define PII pair<int, int>
#define F first
#define S second
#define B begin()
#define E end()
using namespace std;
typedef long long LL;
//head

const int N = 1000010;
int rt, tot;
int fa[N], ch[N][2], data[N], val[N];

inline void clear(int x){ fa[x] = ch[x][0] = ch[x][1] = data[x] = val[x] = 0; }

inline void rot(int x)
{
    int y = fa[x], z = fa[y];
    int f = ch[y][1] == x;
    ch[y][f] = ch[x][f^1]; if (ch[x][f^1]) fa[ch[x][f^1]] = y;
    fa[x] = z; if (z) ch[z][ch[z][1] == y] = x;
    fa[y] = x; ch[x][f^1] = y;
}

inline void splay(int x, int top)
{
    while (fa[x] != top){
        int y = fa[x], z = fa[y];
        if (z != top) rot(((ch[z][0] == y) == (ch[y][0] == x)) ? y : x);
        rot(x);
    } if (!top) rt = x;
}

inline void insert(int K, int v)
{
    int x = rt;
    if (!rt){
        x = rt = ++ tot;
        data[x] = v; val[x] = K;
        fa[x] = ch[x][0] = ch[x][1] = 0;
        return;
    }
    while (x){
        int &y = ch[x][v > data[x]];
        if (!y){
            y = ++ tot;
            data[y] = v; val[y] = K;
            fa[y] = x; ch[y][0] = ch[y][1] = 0;
            x = y; return;
        } x = y;
    } splay(x, 0);
}

inline int findMax(int x){ while (ch[x][1]) x = ch[x][1]; return x; }

inline int findMin(int x){ while (ch[x][0]) x = ch[x][0]; return x; }

inline void del(int x)
{
    splay(x, 0);
    if (!ch[x][0] && !ch[x][1]){ clear(rt); rt = 0; return; }
    if (!ch[x][0]){ rt = ch[x][1]; fa[rt] = 0; return; }
    if (!ch[x][1]){ rt = ch[x][0]; fa[rt] = 0; return; }
    int p = findMax(ch[x][0]), rtt = rt; splay(p, 0);
    ch[rt][1] = ch[rtt][1]; fa[ch[rtt][1]] = rt;
    clear(rtt);
}

int main()
{
    int opt;
    while (~scanf("%d", &opt) && opt){
        if (opt == 1){
            int x, y; scanf("%d%d", &x, &y);
            insert(x, y);
        } else if (opt == 2){
            int x = findMax(rt);
            if (!x){ puts("0"); continue; }
            printf("%d\n", val[x]);
            del(x);
        } else {
            int x = findMin(rt);
            if (!x){ puts("0"); continue; }
            printf("%d\n", val[x]);
            del(x);
        }
    }
    return 0;
}