线段树套 treap / 树状数组套 treap

最新推荐文章于 2022-01-26 23:18:59 发布

是哆啦D梦

最新推荐文章于 2022-01-26 23:18:59 发布

阅读量264

点赞数

文章标签：算法

本文链接：https://blog.csdn.net/weixin_43602607/article/details/120657079

版权

线段树套无旋 treap

前言

线段树套 treap 是最基础的树套树，没有之一。treap 可以实现的功能，例如：第 k 大，比 val 小的数的个数等，使用线段树套 treap 之后，都可以实现在区间中找第 k 大，在区间中找比 val 小的数的个数。并且其 支持单点修改。

算法学习

思路： 非常简单的一个思路：假设线段树维护区间 $[l, r]$ ，那么其就有对应一个 $t r e a p$ 维护区间 $[l, r]$ 之间的所有点。空间复杂度： $n l o g n$

#include<bits/stdc++.h>
using namespace std;
const int N=1e7+10;
int root[N],cnt,R=2e5;
struct node {
	int val,key,l,r,size;
} tr[N];

void pushup(int now) {
	tr[now].size=tr[tr[now].l].size+tr[tr[now].r].size+1;
}
int newnode(int val) {
	tr[++cnt].val=val;
	tr[cnt].l=tr[cnt].r=0;
	tr[cnt].size=1;
	tr[cnt].key=rand();
	return cnt;
}
void split(int now,int &a,int &b,int val) {
	if(now==0) {
		a=b=0;
		return;
	}
	if(tr[now].val<=val)a=now,split(tr[now].r,tr[a].r,b,val);
	else b=now,split(tr[now].l,a,tr[b].l,val);
	pushup(now);
}
void merge(int &now,int a,int b) {//把已 a 为根的树和已 b 为根的树合并为以 now 为根的树
	if(a==0||b==0) {
		now=a+b;
		return;
	}
	if(tr[a].key>tr[b].key)now=a,merge(tr[now].r,tr[a].r,b);
	else now=b,merge(tr[now].l,a,tr[b].l);
	pushup(now);
}
void insert(int &root,int val) {//在root这棵树内插入一个值为val的元素 
	int x=0,y=0;
	split(root,x,y,val-1);
	merge(x,x,newnode(val));
	merge(root,x,y);
}
void build(int now,int l,int r){//其实可以不用build，nlogn update也行
	for(int i=l;i<=r;i++)insert(root[now],a[i]);
	if(l==r)return;
	int mid=(l+r)/2;
	build(now*2,l,mid);
	build(now*2+1,mid+1,r);
}
void update(int now,int l,int r,int x,int val){
	del(root[now],a[x]);//因题而定 
	insert(root[now],val);
	if(l==r)return;
	int mid=(l+r)/2;
	if(x<=mid)update(now*2,l,mid,x,val);
	else update(now*2+1,mid+1,r,x,val);
}
int query(int now,int l,int r,int ql,int qr){
	if(ql<=l&&r<=qr)return findAns(root[now]);
	int mid=(l+r)/2,ans=0;
	if(ql<=mid)ans+=query(now*2,l,mid,ql,qr);
	if(qr>mid)ans+=update(now*2+1,mid+1,r,ql,qr);
	return ans;
}

例题1：三维逆序对数量

题目描述： 有 n 个元素，每个元素都有三个属性 $a_i,b_i,c_i$ 。对于每个 $j$ ，求 $a_i<=a_j$ 且 $b_i<=b_j$ 且 $c_i<=c_j$ 的 $i$ 的个数。 $1<=n,a_i,b_i,c_i<=2e5$ 。数据保证不会有 $a, b, c$ 均相等的元素。
问题分析： 类似二维的树状数组求逆序对。对于三维而言，先对 $a$ 排序，再用线段树套 treap 查区间 $1,b_j]$ 中， $c<=c_j$ 的个数。后者可以用线段树套 treap 做。

#include<bits/stdc++.h>
using namespace std;
const int N=1e7+10;

int root[N],cnt,ans[N],R=2e5;
struct node {
	int val,key,l,r,size;
} tr[N];
struct Elem {
	int a,b,c,id;
} f[N];

void pushup(int now) {
	tr[now].size=tr[tr[now].l].size+tr[tr[now].r].size+1;
}
int newnode(int val) {
	tr[++cnt].val=val;
	tr[cnt].l=tr[cnt].r=0;
	tr[cnt].size=1;
	tr[cnt].key=rand();
	return cnt;
}
void split(int now,int &a,int &b,int val) {
	if(now==0) {
		a=b=0;
		return;
	}
	if(tr[now].val<=val)a=now,split(tr[now].r,tr[a].r,b,val);
	else b=now,split(tr[now].l,a,tr[b].l,val);
	pushup(now);
}
void merge(int &now,int a,int b) {//把已 a 为根的树和已 b 为根的树合并为以 now 为根的树
	if(a==0||b==0) {
		now=a+b;
		return;
	}
	if(tr[a].key>tr[b].key)now=a,merge(tr[now].r,tr[a].r,b);
	else now=b,merge(tr[now].l,a,tr[b].l);
	pushup(now);
}
void insert(int &root,int val) {//在root这棵树内插入一个值为val的元素 
	int x=0,y=0;
	split(root,x,y,val-1);
	merge(x,x,newnode(val));
	merge(root,x,y);
}
int find(int root,int val) {//查找小于等于val的元素个数 
	int x=0,y=0;
	split(root,x,y,val);
	int ans=tr[x].size;
	merge(root,x,y);
	return ans;
}
int cmp(Elem x,Elem y) {
	if(x.a==y.a) {
		if(x.b==y.b)return x.c<y.c;
		else return x.b<y.b;
	} else return x.a<y.a;
}
int query(int now,int l,int r,int ql,int qr,int y) {
	if(ql<=l&&r<=qr)return find(root[now],y);
	int mid=(l+r)/2,ans=0;
	if(ql<=mid)ans+=query(now*2,l,mid,ql,qr,y);
	if(qr>mid)ans+=query(now*2+1,mid+1,r,ql,qr,y);
	return ans;
}
void update(int now,int l,int r,int x,int y) {
	insert(root[now],y);
	if(l==r)return;
	int mid=(l+r)/2;
	if(x<=mid)update(now*2,l,mid,x,y);
	else update(now*2+1,mid+1,r,x,y);
}
int main() {
	int n;
	cin>>n;
	for(int i=1; i<=n; i++)cin>>f[i].a>>f[i].b>>f[i].c,f[i].id=i;
	sort(f+1,f+n+1,cmp);
	for(int i=1; i<=n; i++) {
		ans[f[i].id]=query(1,1,R,1,f[i].b,f[i].c);
		update(1,1,R,f[i].b,f[i].c);
	}
    for(int i=1;i<=n;i++)cout<<ans[i]<<endl;
}

树状数组套无旋 treap

前言

与线段树套 treap 基本一致，甚至可以说：它能实现的功能，线段树套 treap 都可以实现。学它的目的在于：常数小，好写，代码短。

算法学习

思路： 思路与线段树维护区间 $[l, r]$ 一样，对于每个区间 $[l, r]$ ，其就有对应一个 $t r e a p$ 维护区间 $[l, r]$ 之间的所有点。空间复杂度： $n l o g n$

#include<bits/stdc++.h>
using namespace std;
const int N=1e7+10;

int root[N],cnt,ans[N],R=2e5;
struct node {
	int val,key,l,r,size;
} tr[N];

void pushup(int now) {
	tr[now].size=tr[tr[now].l].size+tr[tr[now].r].size+1;
}
int newnode(int val) {
	tr[++cnt].val=val;
	tr[cnt].l=tr[cnt].r=0;
	tr[cnt].size=1;
	tr[cnt].key=rand();
	return cnt;
}
void split(int now,int &a,int &b,int val) {
	if(now==0) {
		a=b=0;
		return;
	}
	if(tr[now].val<=val)a=now,split(tr[now].r,tr[a].r,b,val);
	else b=now,split(tr[now].l,a,tr[b].l,val);
	pushup(now);
}
void merge(int &now,int a,int b) {//把已 a 为根的树和已 b 为根的树合并为以 now 为根的树
	if(a==0||b==0) {
		now=a+b;
		return;
	}
	if(tr[a].key>tr[b].key)now=a,merge(tr[now].r,tr[a].r,b);
	else now=b,merge(tr[now].l,a,tr[b].l);
	pushup(now);
}
void insert(int &root,int val) {//在root这棵树内插入一个值为val的元素 
	int x=0,y=0;
	split(root,x,y,val-1);
	merge(x,x,newnode(val));
	merge(root,x,y);
}
int lowbit(int x){
	return x&(-x);
}
void update(int x,int val){
	for(int i=x;i<=R;i+=lowbit(i))insert(root[i],val);
}
int query(int x){
	int ans=0;
	for(int i=x;i>0;i-=lowbit(i))ans+=findAns(root[i]);
	return ans;
}

例题1：三维逆序对数量

题目描述： 有 n 个元素，每个元素都有三个属性 $a_i,b_i,c_i$ 。对于每个 $j$ ，求 $a_i<=a_j$ 且 $b_i<=b_j$ 且 $c_i<=c_j$ 的 $i$ 的个数。 $1<=n,a_i,b_i,c_i<=2e5$ 。数据保证不会有 $a, b, c$ 均相等的元素。
问题分析： 类似二维的树状数组求逆序对。对于三维而言，先对 $a$ 排序，再用线段树套 treap 查区间 $1,b_j]$ 中， $c<=c_j$ 的个数。后者可以用线段树套 treap 做。

#include<bits/stdc++.h>
using namespace std;
const int N=1e7+10;

int root[N],cnt,ans[N],R=2e5;
struct node {
	int val,key,l,r,size;
} tr[N];
struct Elem {
	int a,b,c,id;
} f[N];

void pushup(int now) {
	tr[now].size=tr[tr[now].l].size+tr[tr[now].r].size+1;
}
int newnode(int val) {
	tr[++cnt].val=val;
	tr[cnt].l=tr[cnt].r=0;
	tr[cnt].size=1;
	tr[cnt].key=rand();
	return cnt;
}
void split(int now,int &a,int &b,int val) {
	if(now==0) {
		a=b=0;
		return;
	}
	if(tr[now].val<=val)a=now,split(tr[now].r,tr[a].r,b,val);
	else b=now,split(tr[now].l,a,tr[b].l,val);
	pushup(now);
}
void merge(int &now,int a,int b) {//把已 a 为根的树和已 b 为根的树合并为以 now 为根的树
	if(a==0||b==0) {
		now=a+b;
		return;
	}
	if(tr[a].key>tr[b].key)now=a,merge(tr[now].r,tr[a].r,b);
	else now=b,merge(tr[now].l,a,tr[b].l);
	pushup(now);
}
void insert(int &root,int val) {//在root这棵树内插入一个值为val的元素 
	int x=0,y=0;
	split(root,x,y,val-1);
	merge(x,x,newnode(val));
	merge(root,x,y);
}
int find(int root,int val) {//查找小于等于val的元素个数 
	int x=0,y=0;
	split(root,x,y,val);
	int ans=tr[x].size;
	merge(root,x,y);
	return ans;
}
int cmp(Elem x,Elem y) {
	if(x.a==y.a) {
		if(x.b==y.b)return x.c<y.c;
		else return x.b<y.b;
	} else return x.a<y.a;
}
int lowbit(int x){
	return x&(-x);
}
void update(int x,int val){
	for(int i=x;i<=R;i+=lowbit(i))insert(root[i],val);
}
int query(int x,int val){
	int ans=0;
	for(int i=x;i>0;i-=lowbit(i))ans+=findAns(root[i],val);
	return ans;
} 
int main() {
	int n;
	cin>>n;
	for(int i=1; i<=n; i++)cin>>f[i].a>>f[i].b>>f[i].c,f[i].id=i;
	sort(f+1,f+n+1,cmp);
	for(int i=1; i<=n; i++) {
		ans[f[i].id]=query(f[i].b,f[i].c);
		update(f[i].b,f[i].c);
	}
    for(int i=1;i<=n;i++)cout<<ans[i]<<endl;
}

是哆啦D梦

关注

0
点赞
踩
1

收藏

觉得还不错? 一键收藏
0
评论
线段树套 treap / 树状数组套 treap

线段树套无旋 treap前言线段树套 treap 是最基础的树套树，没有之一。treap 可以实现的功能，例如：第 k 大，比 val 小的数的个数等，使用线段树套 treap 之后，都可以实现在区间中找第 k 大，在区间中找比 val 小的数的个数。并且其支持单点修改。算法学习思路：非常简单的一个思路：假设线段树维护区间 [l,r][l,r][l,r] ，那么其就有对应一个 treaptreaptreap 维护区间 [l,r][l,r][l,r] 之间的所有点。空间复杂度：nlognn
复制链接

扫一扫