splay树---板子题

NP_hard

已于 2022-01-28 19:14:22 修改

阅读量381

点赞数

分类专栏： Data Structure 文章标签： c++ 深度优先蓝桥杯

于 2020-12-19 22:54:54 首次发布

本文链接：https://blog.csdn.net/NP_hard/article/details/111084593

版权

这篇博客介绍了Splay树的两个应用问题，包括问题A：Splay树的基本操作，以及问题B：宠物收养所中如何使用Splay树进行前驱后继操作。博客提供了样例输入、输出和代码实现，帮助读者理解Splay树在解决实际问题中的应用。

摘要由CSDN通过智能技术生成

文章目录

问题 A: Splay —— Ver.I

题目描述

在这里插入图片描述

输入

第一行包含一个整数n，表示初始序列的长度。以下n行每行包含一个整数，描述初始的序列。接下来一行包含一个整数n，表示插入操作的数目。以下m行每行描述一个操作。

接下来一行包含一个整数q，表示查询和删除操作的总数目，以下q行描述一个操作

输出

对于所有操作，输出正确的答案。

样例输入

5
1 2 3 4 5
3
ADD 9
ADD 10
ADD 199
5
QUERY 1
QUERY 3
DELETE 1
DELETE 3
QUERY 3

样例输出

the number is good
the number is good
2 3 4 5 9 10 199
2 4 5 9 10 199
the number is bad

代码1

#include <cstdio>
#include<iostream>
#include<string>
using namespace std;

const int N = 100005;
int rt, tot, fa[N], ch[N][2], val[N], cnt[N], sz[N];

class tree
{
   
public:
    void maintain(int x){
   
        sz[x] = sz[ch[x][0]] + sz[ch[x][1]] + cnt[x]; 
    }

    bool get(int x){
    
        return x == ch[fa[x]][1]; 
    }

    void clear(int x){
   
        ch[x][0] = ch[x][1] = fa[x] = val[x] = sz[x] = cnt[x] = 0;
    }
    void rotate(int x){
   
        int y = fa[x], z = fa[y], chk = get(x);
        ch[y][chk] = ch[x][chk ^ 1];
        fa[ch[x][chk ^ 1]] = y;
        ch[x][chk ^ 1] = y;
        fa[y] = x;
        fa[x] = z;
        if (z) ch[z][y == ch[z][1]] = x;
        maintain(x);
        maintain(y);
    }
    void splay(int x){
   
        for (int f = fa[x]; f = fa[x], f; rotate(x))
        if (fa[f]) rotate(get(x) == get(f) ? f : x);
        rt = x;
    }
    void insert(int k){
   
        if (!rt) {
   
            val[++tot] = k;
            cnt[tot]++;
            rt = tot;
            maintain(rt);
            return;
        }
        int cnr = rt, f = 0;
        while (1) {
   
            if (val[cnr] == k){
   
                cnt[cnr]++;
                maintain(cnr);
                maintain(f);
                splay(cnr);
                break;
            }
            f = cnr;
            cnr = ch[cnr][val[cnr] < k];
            if (!cnr) {
   
                val[++tot] = k;
                cnt[tot]++;
                fa[tot] = f;
                ch[f][val[f] < k] = tot;
                maintain(tot);
                maintain(f);
                splay(tot);
                break;
            }
        }
    }
    int rk(int k) {
   
        int res = 0, cnr = rt;
        while (1) {
   
            if (k < val[cnr]) {
   
                cnr = ch[cnr][0];
            } else {
   
                res += sz[ch[cnr][0]];
                if (k == val[cnr]) {
   
                splay(cnr);
                return res + 1;
                }
                res += cnt[cnr];
                cnr = ch[cnr][1];
            }
        }
    }
    int kth(int k) {
   
        int cnr = rt;
        while (1) {
   
            if (ch[cnr][0] && k <= sz[ch[cnr][0]]) {
   
                cnr = ch[cnr][0];
            } else {
   
                k -= cnt[cnr] + sz[ch[cnr][0]];
                if (k <= 0) {
   
                splay(cnr);
                return val[cnr];
                }
                cnr = ch[cnr][1];
            }
        }
    }
    int <