洛谷 P2286 [HNOI2004]宠物收养场 (treap 树)

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 + 
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值