treap 树
本题要点:
1、建立两棵 treap 树, 一棵用来存 宠物,一棵树用来存 顾客。 题目给出的所有的顾客值和宠物值都不一样。
这样结构体 Treap 只需要 多一个 size 来就来子树的节点总数
2、套用 treap 树 的模板,写好 以下函数。 因为是两棵树,需要传参数的引用。
int New(int val, Treap* a, int& tot)
void update(int p, Treap* a)
void Build(Treap* a, int& tot, int& root)
void zig(int &p, Treap* a) //右旋转
void zag(int &p, Treap* a) //左旋
void Insert(int &p, int val, Treap* a, int& tot)
int GetPre(int val, Treap* a, int root)
int GetNext(int val, Treap* a, int root)
void Remove(int &p, int val, Treap* a)
3、如果新来了宠物 val ,那么判断顾客的 树是否为空,
如果顾客为空,把 val 插入到 宠物的 treap树;
否则,从顾客树中查找前驱 pre, 后继 nxt, 判断差值,看看选哪个。选了再删除。 还有,这里优先选择前驱。
如果新来的是 顾客 val, 同理操作。
#include <cstdio>
#include <cstring>
#include <iostream>
#include <cstdlib>
using namespace std;
const int MaxN = 8e4 + 10, INF = 0x3fffffff, mod = 1000000;
int n, tot1, tot2, root1, root2;
struct Treap
{
int l, r;
int val, dat;
int size; // 子树的节点总数
}client[MaxN], pet[MaxN];
int New(int val, Treap* a, int& tot)
{
a[++tot].val = val;
a[tot].dat = rand();
a[tot].size = 1;
return tot;
}
void update(int p, Treap* a)
{
a[p].size = a[a[p].l].size + a[a[p].r].size +