splay板子整理

本质就是个平衡树。
具体到一道题,无非就是要考虑如何
1.每个节点维护什么信息。
2.如何维护splay(top,bottom等splay特色操作)
3.如何建树,建树规则如何实现。如何找到需要的节点。
4.结合其他数据结构(难)

#include <bits/stdc++.h>

using namespace std;
//-----pre_def----
const double PI = acos(-1.0);
const int INF = 0x3f3f3f3f;
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int, int> PII;
typedef pair<double, double> PDD;
#define fir(i, a, b) for (int i = (a); i <= (b); i++)
#define rif(i, a, b) for (int i = (a); i >= (b); i--)
#define endl '\n'
#define init_h memset(h, -1, sizeof h), idx = 0;
#define lowbit(x) x &(-x)

//---------------
const int N = 1e5 + 10;

struct node //节点信息
{
    int s[2], p, v, sz;
    //int lz;
} tr[N];
int n, m, root, idx;
void pushup(int u)
{
    //维护父子之间的信息
    tr[u].sz = tr[tr[u].s[0]].sz + tr[tr[u].s[1]].sz + 1;
}
void rotate(int x)
{
    int y = tr[x].p, z = tr[y].p; //在这个链上:z是祖宗,y是爸爸,x是儿子。现在要把x移动到最上边
    int k = tr[y].s[1] == x;      // k=0表示x是y的左儿子;k=1表示x是y的右儿子
    tr[z].s[tr[z].s[1] == y] = x;
    tr[x].p = z;
    tr[y].s[k] = tr[x].s[k ^ 1];
    tr[tr[x].s[k ^ 1]].p = y;
    tr[x].s[k ^ 1] = y;
    tr[y].p = x;
    pushup(y), pushup(x);
}
void splay(int x, int k = 0)
{
    while (tr[x].p != k)
    {
        int y = tr[x].p, z = tr[y].p;
        if (z != k)
        {
            if ((tr[y].s[1] == x) ^ (tr[z].s[1] == y))
                rotate(x);
            else
                rotate(y);
        }
        rotate(x);
    }
    if (!k)
        root = x;
}
void print(int x = root)
{
    if (tr[x].s[0])
        print(tr[x].s[0]);
    //printf("%lld ", tr[x].value);
    if (tr[x].s[1])
        print(tr[x].s[1]);
}

int insert(int v) //返回节点编号;对于逻辑比较复杂的情况,可以写成递归形式
//对于非数值的情况(比如维护中序遍历)无非是将新节点插至最末尾,然后在splay上来。
{
    int u = root, p = 0;
    while (u)
    {
        p = u, u = tr[u].s[v > tr[u].v];
    }
    u = ++idx;
    if (p)
        tr[p].s[v > tr[p].v] = u;

    tr[u].p = p;
    tr[u].v = v;
    tr[u].sz = 1;
    splay(u);
    return u;
}
int find(int v) //查找v的位置,并将其旋转到根节点(此时左子树的size就是v的排名)
{
    int u = root;
    if (!u)
        return -1;                               //树空
    while (tr[u].s[v > tr[u].v] && v != tr[u].v) //当存在儿子并且当前位置的值不等于x
        u = tr[u].s[v > tr[u].v];                //跳转到儿子,查找x的父节点
    splay(u, 0);                                 //把当前位置旋转到根节点
    return root;
}
int find_pre(int v) //查找v的前驱的节点编号
{
    find(v);
    int u = root; //根节点,此时x的父节点(存在的话)就是根节点
    if (tr[u].v < v)
        return u;   //如果当前节点的值小于x并且要查找的是前驱
    u = tr[u].s[0]; //查找后继的话在右儿子上找,前驱在左儿子上找
    while (tr[u].s[1])
        u = tr[u].s[1]; //要反着跳转,否则会越来越大(越来越小)
    return u;           //返回位置
}
int find_ne(int v) //查找v的后继的节点编号
{
    find(v);
    int u = root; //根节点,此时x的父节点(存在的话)就是根节点
    if (tr[u].v > v)
        return u;   //如果当前节点的值小于x并且要查找的是前驱
    u = tr[u].s[1]; //查找后继的话在右儿子上找,前驱在左儿子上找
    while (tr[u].s[0])
        u = tr[u].s[0]; //要反着跳转,否则会越来越大(越来越小)
    return u;           //返回位置
}
void del(int v) //删除值为v的点
{
    int last = find_pre(v); //查找x的前驱
    int next = find_ne(v);  //查找x的后继
    splay(last, 0);
    splay(next, last);
    //将前驱旋转到根节点,后继旋转到根节点下面
    //很明显,此时后继是前驱的右儿子,x是后继的左儿子,并且x是叶子节点
    int del = tr[next].s[0]; //后继的左儿子
    tr[next].s[0] = 0;       //这个节点直接丢掉(不存在了)
}
int get(int v)
{
    int u = root, res;
    while (u)
    {
        if (tr[u].v >= v)
            res = u, u = tr[u].s[0];
        else
            u = tr[u].s[1];
    }
    return res;
}
void del(int l, int r) //删除值为l~r的所有点
{
    int last = get(l); //查找x的前驱
    int next = get(r); //查找x的后继
    splay(next, 0);
    splay(last, next);
    tr[last].s[1] = 0; //这个节点直接丢掉(不存在了)
    pushup(last), pushup(next);
}
int find_k(int k) //查找排名为k的数
{
    int u = root;
    while (u)
    {
        if (tr[tr[u].s[0]].sz >= k)
            u = tr[u].s[0];
        else if (tr[tr[u].s[0]].sz + 1 == k)
            return tr[u].v;
        else
            k -= tr[tr[u].s[0]].sz + 1, u = tr[u].s[1];
    }
    return -1;
}
void init()
{
    insert(INF);
    insert(-INF);
}
int main()
{
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
    int StartTime = clock();
#endif
    int T = 1;
    scanf("%d", &T);
    while (T--)
    {
    }
#ifndef ONLINE_JUDGE
    printf("Run_Time = %d ms\n", clock() - StartTime);
#endif
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值