【bzoj3224】普通平衡树 平衡树

最新推荐文章于 2019-07-26 23:06:59 发布
最新推荐文章于 2019-07-26 23:06:59 发布
阅读量450 收藏
点赞数
分类专栏： bzoj 平衡树
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/chty2018/article/details/53838146
版权

AC通道：http://www.lydsy.com/JudgeOnline/problem.php?id=3224

借此题来练习平衡树的模板。（以下数据均为bzoj亲测数据）

【Treap】

Memory：3762kb

Time：248ms  

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
using namespace std;
#define FILE "read"
#define MAXN 100005
int n,len,root,ans;
namespace INIT{
	char buf[1<<15],*fs,*ft;
	inline char getc() {return (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<15,stdin),fs==ft))?0:*fs++;}
	inline int read(){
		int x=0,f=1;  char ch=getc();
		while(!isdigit(ch))  {if(ch=='-')  f=-1;  ch=getc();}
		while(isdigit(ch))  {x=x*10+ch-'0';  ch=getc();}
		return x*f;
	}
}using namespace INIT;
namespace Treap_Tree{
	struct node{int l,r,v,size,fix,w;}tr[MAXN];
	void update(int k){tr[k].size=tr[tr[k].l].size+tr[tr[k].r].size+tr[k].w;}
	void rturn(int &k){int t=tr[k].l;tr[k].l=tr[t].r;tr[t].r=k;tr[t].size=tr[k].size;update(k);k=t;}
	void lturn(int &k){int t=tr[k].r;tr[k].r=tr[t].l;tr[t].l=k;tr[t].size=tr[k].size;update(k);k=t;}
	void insert(int &k,int x){
		if(k==0){len++;k=len;tr[k].size=tr[k].w=1;tr[k].v=x;tr[k].fix=rand();return;}
		tr[k].size++;
		if(tr[k].v==x)tr[k].w++;
		else if(x>tr[k].v){insert(tr[k].r,x);if(tr[tr[k].r].fix<tr[k].fix)lturn(k);}
		else {insert(tr[k].l,x);if(tr[tr[k].l].fix<tr[k].fix)rturn(k);} 
	}
	void del(int &k,int x){
		if(k==0)return; 
		if(tr[k].v==x){
			if(tr[k].w>1){tr[k].w--;tr[k].size--;return;}
			if(tr[k].l*tr[k].r==0)k=tr[k].l+tr[k].r;
			else if(tr[tr[k].l].fix<tr[tr[k].r].fix)rturn(k),del(k,x);
			else lturn(k),del(k,x);
		}
		else if(x>tr[k].v)tr[k].size--,del(tr[k].r,x);
		else tr[k].size--,del(tr[k].l,x);
	}
	int rank(int k,int x){
		if(k==0)return 0;
		if(tr[k].v==x)return tr[tr[k].l].size+1;
		else if(x>tr[k].v)return tr[tr[k].l].size+tr[k].w+rank(tr[k].r,x);
		else return rank(tr[k].l,x);
	}
	int Findkth(int k,int x){
		if(k==0)return 0;
		if(x<=tr[tr[k].l].size)return Findkth(tr[k].l,x);
		else if(x>tr[tr[k].l].size+tr[k].w)return Findkth(tr[k].r,x-tr[tr[k].l].size-tr[k].w);
		else return tr[k].v;
	}
	void get_before(int k,int x){
		if(k==0)return;
		if(tr[k].v<x){ans=k;get_before(tr[k].r,x);}
		else get_before(tr[k].l,x);
	}
	void get_behind(int k,int x){
		if(k==0)return;
		if(tr[k].v>x){ans=k;get_behind(tr[k].l,x);}
		else get_behind(tr[k].r,x);
	}
}using namespace Treap_Tree;
int main(){
	freopen(FILE".in","r",stdin);
	freopen(FILE".out","w",stdout);
	n=read();
	int flag,x;
	for(int i=1;i<=n;i++){
		flag=read();  x=read();
		switch(flag){
			case 1:insert(root,x);break;
			case 2:del(root,x);break;
			case 3:printf("%d\n",rank(root,x));break;
			case 4:printf("%d\n",Findkth(root,x));break;
			case 5:ans=0;get_before(root,x);printf("%d\n",tr[ans].v);break;
			case 6:ans=0;get_behind(root,x);printf("%d\n",tr[ans].v);break;
		}
	}
	return 0;
}
【Splay】

Memory:3668kb

Time:484ms

/*************
  Splay模板
  by chty
  2016.12.23
*************/
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
using namespace std;
#define FILE "phs"
#define MAXN 100010
#define up(i,j,n) for(int i=j;i<=n;i++)
#define dn(i,j,n) for(int i=j;i>=n;i--)
#define cmax(a,b) a=max(a,b)
#define cmin(a,b) a=min(a,b)
namespace INIT{
	char buf[1<<15],*fs,*ft;
	inline char getc() {return (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<15,stdin),fs==ft))?0:*fs++;}
	inline int read(){
		int x=0,f=1;  char ch=getc();
		while(!isdigit(ch))  {if(ch=='-')  f=-1;  ch=getc();}
		while(isdigit(ch))  {x=x*10+ch-'0';  ch=getc();}
		return x*f;
	}
}using namespace INIT;
namespace Splay_Tree{
	int root,cnt,v[MAXN],w[MAXN],size[MAXN],f[MAXN],son[MAXN][2];
	bool get(int x) {return son[f[x]][1]==x;}
	void clear(int x) {son[x][0]=son[x][1]=v[x]=w[x]=f[x]=size[x]=0;}
	void updata(int x) {size[x]=size[son[x][0]]+size[son[x][1]]+w[x];}
	int before(){int now=son[root][0];while(son[now][1])  now=son[now][1];return now;}
	int behind(){int now=son[root][1];while(son[now][0])  now=son[now][0];return now;}
	void rotate(int x){
		int y=f[x],z=f[y],which=get(x);
		son[y][which]=son[x][which^1];  f[son[y][which]]=y;
		f[y]=x;  son[x][which^1]=y;
		f[x]=z;  if(z)  son[z][son[z][1]==y]=x;
		updata(y);  updata(x);
	}
	void splay(int x){for(int fa;(fa=f[x]);rotate(x))if(f[fa])rotate((get(x)==get(fa)?fa:x));root=x;}
	void insert(int value){
		int now=root,fa(0);
		if(!now) {root=++cnt;v[root]=value;size[root]=w[root]=1;return;}
		while(1){
			if(value==v[now]) {w[now]++;splay(now);break;}
			fa=now; now=son[now][value>v[now]];
			if(now==0){
				now=++cnt;v[now]=value;size[now]=w[now]=1;
				f[now]=fa; son[fa][value>v[fa]]=now;splay(now); 
				break;
			}
		}
	}
	int find_rank(int value){
		int now=root,ans(0);
		while(1){
			if(value<v[now])  now=son[now][0];
			else{
				ans+=size[son[now][0]];
				if(value==v[now]) {splay(now); return ans+1;}
				ans+=w[now];
				now=son[now][1];
			}
		}
	}
	int find_Kth(int x){
		int now=root;
		while(1){
			if(son[now][0]&&x<=size[son[now][0]])  now=son[now][0];
			else{
				int temp=size[son[now][0]]+w[now];
				if(x<=temp)  return v[now];
				x-=temp;  now=son[now][1];
			}
		}
	}
	void del(int value){
		find_rank(value);
		if(w[root]>1)  {w[root]--; size[root]--;}
		else if(son[root][0]+son[root][1]==0) clear(root),root=0;
		else if(!son[root][0]) {int last=root;root=son[root][1];f[root]=0;clear(last);}
		else if(!son[root][1]) {int last=root;root=son[root][0];f[root]=0;clear(last);}
		else{
			int x=before(),last=root;
			splay(x);  son[root][1]=son[last][1];  f[son[last][1]]=root;
			clear(last);  updata(root);  
		}
	}
}using namespace Splay_Tree;
int main(){
	freopen(FILE".in","r",stdin);
	freopen(FILE".out","w",stdout);
	int n=read();
	for(int i=1;i<=n;i++){
		int flag=read(),x=read();
		switch(flag){
			case 1:insert(x);break;
			case 2:del(x);break;
			case 3:printf("%d\n",find_rank(x));break;
			case 4:printf("%d\n",find_Kth(x));break;
			case 5:insert(x);printf("%d\n",v[before()]);del(x);break;
			case 6:insert(x);printf("%d\n",v[behind()]);del(x);break;
		}
	}
	return 0;
}

【替罪羊树】

Memory：3668kb

Time：180ms

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<ctime>
#include<algorithm>
using namespace std;
#define INF 1000000000
#define FILE "read"
int n,len,root,top,stack[100005];
namespace INIT{
	char buf[1<<15],*fs,*ft;
	inline char getc() {return (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<15,stdin),fs==ft))?0:*fs++;}
	inline int read(){
		int x=0,f=1;  char ch=getc();
		while(!isdigit(ch))  {if(ch=='-')  f=-1;  ch=getc();}
		while(isdigit(ch))  {x=x*10+ch-'0';  ch=getc();}
		return x*f;
	}
}using namespace INIT;
namespace TiZuiYang_Tree{
	const double chty=0.75;  //平衡常数
	struct node{int son[2],f,size,v;}tr[100005];
	void init(){
		len=2;  root=1;
		tr[1].size=2;  tr[1].v=-INF;  tr[1].son[1]=2;
		tr[2].size=1;  tr[2].v=INF;   tr[2].f=1;
	}
	bool balance(int x) {
		double p=tr[x].size*chty;
		return p>=(double)tr[tr[x].son[0]].size&&p>=(double)tr[tr[x].son[1]].size;
	}
	void dfs(int x){//中序遍历
		if(!x)  return;
		dfs(tr[x].son[0]);
		stack[++top]=x;
		dfs(tr[x].son[1]);
	}
	int build(int l,int r){
		if(l>r)  return 0;
		int mid=(l+r)/2,x=stack[mid];
		tr[tr[x].son[0]=build(l,mid-1)].f=x;
		tr[tr[x].son[1]=build(mid+1,r)].f=x;
		tr[x].size=tr[tr[x].son[0]].size+tr[tr[x].son[1]].size+1;
		return x;
	}
	void rebuild(int x){
		top=0;  dfs(x);
		int fa=tr[x].f,which=(tr[fa].son[1]==x);
		int sonroot=build(1,top);
		tr[tr[fa].son[which]=sonroot].f=fa;
		if(x==root)  root=sonroot;
	}
	int find(int v){
		int now=root;
		while(now){
			if(v==tr[now].v)  return now;
			else now=tr[now].son[v>tr[now].v];
		}
	}
	void insert(int v){
		int now=root,p=++len; 
		tr[p].v=v;  tr[p].size=1;//新开结点
		while(1){
			tr[now].size++;
			int which=(v>=tr[now].v);//表示p在当前根的哪个子树内
			if(tr[now].son[which])  now=tr[now].son[which];  
			else {tr[tr[now].son[which]=p].f=now;  break;}
		}
		int id=0;
		for(int i=p;i;i=tr[i].f)  if(!balance(i))  id=i;//记录不平衡点
		if(id)  rebuild(id);//重构子树
	}
	void del(int x){
		if(tr[x].son[0]&&tr[x].son[1]){
			int p=tr[x].son[0];
			while(tr[p].son[1])  p=tr[p].son[1];
			tr[x].v=tr[p].v;  x=p;
		}
		int Son=(tr[x].son[0])?tr[x].son[0]:tr[x].son[1],which=(tr[tr[x].f].son[1]==x);
		tr[tr[tr[x].f].son[which]=Son].f=tr[x].f;
		for(int i=tr[x].f;i;i=tr[i].f)  tr[i].size--;
		if(x==root)  root=Son;
	}
	int rank(int v){
		int now=root,ans=0;
		while(now){
			if(tr[now].v<v)  {ans+=tr[tr[now].son[0]].size+1;  now=tr[now].son[1];}
			else now=tr[now].son[0];
		}
		return ans;
	}
	int kth(int x){
		int now=root;
		while(1){
			if(tr[tr[now].son[0]].size==x-1)  return now;
			else if(tr[tr[now].son[0]].size>=x)  now=tr[now].son[0];
			else x-=tr[tr[now].son[0]].size+1,now=tr[now].son[1];
		}
		return now;
	}
	int before(int v){
		int now=root,ans=-INF;
		while(now){
			if(tr[now].v<v)  ans=max(ans,tr[now].v),now=tr[now].son[1];
			else now=tr[now].son[0];
		}
		return ans;
	}
	int after(int v){
		int now=root,ans=INF;
		while(now){
			if(tr[now].v>v)  ans=min(ans,tr[now].v),now=tr[now].son[0];
			else now=tr[now].son[1];
		}
		return ans;
	}
}using namespace TiZuiYang_Tree;
int main(){
    freopen(FILE".in","r",stdin);
    freopen(FILE".out","w",stdout);
    init();
    n=read();
    for(int i=1;i<=n;i++){
        int flag=read(),x=read();
        if(flag==1)  insert(x);
        if(flag==2)  del(find(x));
        if(flag==3)  printf("%d\n",rank(x));
        if(flag==4)  printf("%d\n",tr[kth(x+1)].v);
        if(flag==5)  printf("%d\n",before(x));
        if(flag==6)  printf("%d\n",after(x));
    }
    return 0;
}




chty_syq
关注
专栏目录
【bzoj3224】普通平衡树
zyf2000
02-04 4935
如何让你遇见我，在我最美丽的时刻。
【BZOJ3224】【TYVJ1728】普通平衡树
cz_xuyixuan的博客
01-14 635
【题目链接】 点击打开链接 【思路要点】 本题包含了平衡树最基本的操作。是任何学习平衡树都应当先做一遍的题。笔者实现了四种平衡树，Splay、Treap、替罪羊树和非旋转式Treap（以及其可持久化）。 【代码】 Splay/*Splay Tree Version*/ #include using namespace std;
bzoj 3224（平衡树）
Chen.Jr的博客
03-21 219
传送门 题意： 请你完成666种最基础的平衡树的操作。 题目分析： 什么也别说了，平衡树的板子题。收获到了一个写起来比较舒服的平衡树的板子。 代码： #include <bits/stdc++.h> using namespace std; const int maxn=200005; int ch[maxn][2],par[maxn],val[maxn],cnt[maxn],siz...
BZOJ3224:普通平衡树(含SBT、Treap、Splay模板)
weixin_34024034的博客
11-22 120
传送门 这道题就是裸的平衡树的题目，操作都是模板，也就不多说了，重在代码。 SBT: #include<cstdio> struct SBT { int v, sz, ch[2], cnt; } t[300005]; int n, cnt, root; #define Upd(k) {\ t[k].sz = t[t[k].ch[0]].sz...
【bzoj3224】普通平衡树 && 平衡树大测速【平衡树】
weixin_30267785的博客
01-12 152
题目链接 若专门来看平衡树测速，请略过这一部分，迅速往下！ GDKOI将近，蒟蒻的博主来复习平衡树怎么打了。 很伤心的是，开心地写了5种解法后，很多没有1A。看来比赛中对拍还是很重要的。只打了自己会的，非旋Treap什么的早就忘到九霄云外去了！ 1 Treap #include<cstdio> #include<cstdlib&gt...
BZOJ3224 普通平衡树(splay)
glk__'blog
07-16 1645
题目在这里 题意：让你实现一棵树，实现 插入， 删除，查询x数的排名，查询排名为x的数 ，求x的前驱(前驱定义为小于x，且最大的数)， 求x的后继(后继定义为大于x，且最小的数) 这道题最开始我用treap过了，今天打了一个splay题解。 treap题解 #include #include #include using namespace std; names
bzoj 3224: Tyvj 1728 普通平衡树
lqybzx的专栏
11-05 682
Description 您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作： 1. 插入x数 2. 删除x数(若有多个相同的数，因只删除一个) 3. 查询x数的排名(若有多个相同的数，因输出最小的排名) 4. 查询排名为x的数 5. 求x的前驱(前驱定义为小于x，且最大的数) 6. 求x的后继(后继定义为大于x，且最小的数) Input 第
红黑树接口(pb_ds)水过普通平衡树BZOJ-3224
weixin_43533544的博客
06-03 908
最近学平衡树，偶然发现了pbds这个库，里面有很多封装的很复杂的数据结构，各种平衡树，可并堆等。这篇文章主要讲一讲，平衡树套pb_ds这个模板的问题 首先，我们要 include 一下我们的主角 #include<bits/stdc++.h> #include<bits/extc++.h> using namespace std; using namespace __g...
区间翻转 bzoj 3223 文艺平衡树 （splay）
w4149
05-26 1932
【bzoj3223】Tyvj 1729 文艺平衡树Description 您需要写一种数据结构（可参考题目标题），来维护一个有序数列，其中需要提供以下操作：翻转一个区间，例如原有序序列是5 4 3 2 1，翻转区间是[2,4]的话，结果是5 2 3 4 1Input 第一行为n,m n表示初始序列有n个数，这个序列依次是(1,2……n-1,n) m表示翻转操作次数 接下来m行每行两个数[l,r
bzoj3224
a5199519的博客
11-20 75
题解： 平衡树模板，打了大半个晚上，调了大半个晚上 都是平衡树的基本操作 代码： #include<bits/stdc++.h> const int N=800005; int root,tot,n,p,q,pre[N],c[N][2],size[N],data[N]; void rot(int x) { int y=pre[x],k=(c[y]...
bzoj3224(普通平衡树 splay板子)
qq_42576687的博客
07-26 167
题目 splay #include<cstdio> #include<iostream> using namespace std; const int N=1e5+5; int ch[N][2],f[N],sz[N],cnt[N],val[N]; int tot,root; inline void _clear(int x){ ch[x][0]=ch[x][1]=...
普通平衡树(分块版)
weixin_30636089的博客
01-21 76
(题面都一样，就不贴了) 今天讲了分块，于是就用分块写了一下这道平衡树模板题hh 离散化之后按权值分块即可，跑得竟然比我平衡树的代码还快hhh /************************************************************** Problem: 3224 User: JYYHH Language: C++ ...
bzoj 3224 普通平衡树(fhqtreap)
Timeclimber
07-27 412
3224: Tyvj 1728 普通平衡树 Time Limit: 10 Sec  Memory Limit: 128 MB Submit: 21374  Solved: 9545 [Submit][Status][Discuss] Description 您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作： 1. 插入x数 2. 删除x数(若有多个相同的数，因只删除...
BZOJ 3224 普通平衡树
我是谁 --- 卡比兽
09-26 187
BZOJ 3224 懒羊羊看了下BST treap  然后对着敲一下BZOJ 3224 代码有注释 额 感觉treap好玄学啊 rand是什么鬼 #include &lt;cstdio&gt; #include &lt;iostream&gt; #include &lt;queue&gt; #include &lt;cmath&gt; #include &lt;cstring&gt; #...
【BZOJ3224】普通平衡树
Facico的博客
07-28 968
Description 您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作： 1. 插入x数 2. 删除x数(若有多个相同的数，因只删除一个) 3. 查询x数的排名(若有多个相同的数，因输出最小的排名) 4. 查询排名为x的数 5. 求x的前驱(前驱定义为小于x，且最大的数) 6. 求x的后继(后继定义为大于x，且最小的数) Solution 这是一道练习
bzoj3224（treap）
cq_phqg
09-02 1103
3224: Tyvj 1728 普通平衡树 Time Limit: 10 Sec  Memory Limit: 128 MB Submit: 1844  Solved: 727 [Submit][Status] Description 您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作： 1. 插入x数 2. 删除x数(若有多个相同的数，因只删除一个)
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值