K-D tree入门更新ing

Posted Mychael

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了K-D tree入门更新ing相关的知识,希望对你有一定的参考价值。

久仰K-D tree大名已久,终于在合适的时候遇见了合适的水题入了坑入了门

K-D tree是什么##

K-D tree是什么? 按名字上翻译来就是K维的树,就是一个用来维护K维空间的点的平衡二叉树

K-D tree有什么用##

K-D tree可以进行空间上的操作,最经典的就是查询最近/最远 点对
还有很多我不知道

K-D tree的原理与实现##

K-D tree,又有一个名字叫做划分树,与其原理相联系

类似于普通的平衡树,对于普通的平衡树的节点u,其左右子树分别是权值小于u的和权值大于u的,该节点就相当于在一个值域上在u的值处进行了分割

而kd-tree对于多维空间进行分割,一个节点储存着以下信息:

struct node{
    int d[K],s[2],x[2],y[2],......;
}e[maxn];

d就是该节点储存的点的K维坐标
s储存着其左右儿子
剩余的若干数组储存着以该节点为根的子树的各维度的最值
也就是说,一棵子树实际上对应着一个空间区域,而根节点将该空间区域划分为左右两部分
而该空间的信息就储存在子树的根节点中

但是这是在一个多维空间,将空间区域划分有多种方式
一般地,kd-tree垂直于其中一个坐标轴将平面划分开
以2维为例,如下图所示

如图圆圈表示节点,将区域进行划分
一般遵循以下规律构造:
①各层节点交替划分各维空间
根节点划分x坐标
其儿子划分y坐标
其孙子划分z坐标
......

②每一层的区域中,按该层划分的坐标排序,选取其中位数作为划分点进行划分
切点作为父节点,左边的点划分到左子树中,右边的点划分到右子树中

③逐层划分,直至划分区域无节点

在C++的STL中,有一个函数nth_element()可以在\\(O(n)\\)时间内将一个数组第k大找出,并将小于的放在左边,大于的放在右边
具体实现类似快排
对于K维空间,建树复杂度\\(O(Knlogn)\\)
二维建树代码如下:

#define ls e[u].s[0]
#define rs e[u].s[1]
#define cmin(x,y) (x > y ? x = y : x)
#define cmax(x,y) (x < y ? x = y : x)
struct point{int d[2];}a[maxn];
struct node{int d[2],s[2],x[2],y[2];}e[maxn];
int n,rt,D,x,y;
bool operator <(const point& a,const point& b){
	return a.d[D] == b.d[D] ? a.d[D ^ 1] < b.d[D ^ 1] : a.d[D] < b.d[D];
}
void pup(int u){
	if (ls){
		cmin(e[u].x[0],e[ls].x[0]); cmax(e[u].x[1],e[ls].x[1]);
		cmin(e[u].y[0],e[ls].y[0]); cmax(e[u].y[1],e[ls].y[1]);
	}
	if (rs){
		cmin(e[u].x[0],e[rs].x[0]); cmax(e[u].x[1],e[rs].x[1]);
		cmin(e[u].y[0],e[rs].y[0]); cmax(e[u].y[1],e[rs].y[1]);
	}
}
int build(int l,int r,int d){
	D = d; int u = l + r >> 1;
	nth_element(a + l,a + u,a + r + 1);
	e[u].d[0] = e[u].x[0] = e[u].x[1] = a[u].d[0];
	e[u].d[1] = e[u].y[0] = e[u].y[1] = a[u].d[1];
	if (l < u) ls =  build(l,u - 1,d ^ 1);
	if (r > u) rs =  build(u + 1,r,d ^ 1);
	pup(u);
	return u;
}

查询最近/远点对##

以最近为例
与普通的暴力不同,在KDtree中查询最近点对,预期复杂度为\\(O(logn)\\),可以被卡为\\(O(\\sqrt{N})\\)
我们到一个节点时,用该节点更新答案,并计算左右子树的估价函数

由于每棵子树都对应一个区域,可以由此计算出每棵子树的最近值
如果最近的点的贡献都比当前答案大,那么就不用访问该子树了
以2维为例,估价函数可以这样写:

#define getd(u) (max(x - e[u].x[1],0) + max(e[u].x[0] - x,0) + max(y - e[u].y[1],0) + max(e[u].y[0] - y,0))
#define getdx(u) (max(abs(e[u].x[0] - x),abs(e[u].x[1] - x)) + max(abs(e[u].y[0] - y),abs(e[u].y[1] - y)))

实际上就是求与四个顶点距离的最值

由此可以写出搜索函数:

void qmx(int u){
	LL t = equal(u) ? -INF : (abs(e[u].d[0] - x) + abs(e[u].d[1] - y)),d[2];
	if (ls) d[0] = getdx(ls); else d[0] = -INF;
	if (rs) d[1] = getdx(rs); else d[1] = -INF;
	cmax(mx,t); t = d[0] <= d[1];
	if (d[t] > mx) qmx(e[u].s[t]); t ^= 1;
	if (d[t] > mx) qmx(e[u].s[t]);
}
void qmn(int u){
	int t = equal(u) ? INF : (abs(e[u].d[0] - x) + abs(e[u].d[1] - y)),d[2];
	if (ls) d[0] = getd(ls); else d[0] = INF;
	if (rs) d[1] = getd(rs); else d[1] = INF;
	cmin(mn,t); t = d[0] >= d[1];
	if (d[t] < mn) qmn(e[u].s[t]); t ^= 1;
	if (d[t] < mn) qmn(e[u].s[t]);
}

插入##

类似平衡树插入

void insert(int u,int d){
	if (e[u].d[d] < dd[d]){
		if (ls) insert(ls,d ^ 1);
		else {
			ls = ++n;
			e[ls].d[0] = e[ls].x[0] = e[ls].x[1] = dd[0];
			e[ls].d[1] = e[ls].y[0] = e[ls].y[1] = dd[1];
		}
	}
	else {
		if (rs) insert(rs,d ^ 1);
		else {
			rs = ++n;
			e[rs].d[0] = e[rs].x[0] = e[rs].x[1] = dd[0];
			e[rs].d[1] = e[rs].y[0] = e[rs].y[1] = dd[1];
		}
	}
	pup(u);
}

例题##

由以上基础,我们就可以轻松A掉SDOI2010 hideseek
用每个点搜一次就好

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define LL long long int
#define Redge(u) for (int k = h[u],to; k; k = ed[k].nxt)
#define REP(i,n) for (int i = 1; i <= (n); i++)
#define BUG(s,n) for (int i = 1; i <= (n); i++) cout<<s[i]<<\' \'; puts("");
#define ls e[u].s[0]
#define rs e[u].s[1]
#define cmin(x,y) (x > y ? x = y : x)
#define cmax(x,y) (x < y ? x = y : x)
#define getd(u) (max(x - e[u].x[1],0) + max(e[u].x[0] - x,0) + max(y - e[u].y[1],0) + max(e[u].y[0] - y,0))
#define getdx(u) (max(abs(e[u].x[0] - x),abs(e[u].x[1] - x)) + max(abs(e[u].y[0] - y),abs(e[u].y[1] - y)))
#define equal(u) (e[u].d[0] == x && e[u].d[1] == y)
using namespace std;
const int maxn = 100005,maxm = 100005,INF = 2100000000;
inline int read(){
	int out = 0,flag = 1; char c = getchar();
	while (c < 48 || c > 57){if (c == \'-\') flag = -1; c = getchar();}
	while (c >= 48 && c <= 57){out = (out << 3) + (out << 1) + c - 48; c = getchar();}
	return out * flag;
}
struct point{int d[2];}a[maxn];
struct node{int d[2],s[2],x[2],y[2];}e[maxn];
int n,rt,D,x,y; LL mx,mn;
bool operator <(const point& a,const point& b){
	return a.d[D] == b.d[D] ? a.d[D ^ 1] < b.d[D ^ 1] : a.d[D] < b.d[D];
}
void pup(int u){
	if (ls){
		cmin(e[u].x[0],e[ls].x[0]); cmax(e[u].x[1],e[ls].x[1]);
		cmin(e[u].y[0],e[ls].y[0]); cmax(e[u].y[1],e[ls].y[1]);
	}
	if (rs){
		cmin(e[u].x[0],e[rs].x[0]); cmax(e[u].x[1],e[rs].x[1]);
		cmin(e[u].y[0],e[rs].y[0]); cmax(e[u].y[1],e[rs].y[1]);
	}
}
int build(int l,int r,int d){
	D = d; int u = l + r >> 1;
	nth_element(a + l,a + u,a + r + 1);
	e[u].d[0] = e[u].x[0] = e[u].x[1] = a[u].d[0];
	e[u].d[1] = e[u].y[0] = e[u].y[1] = a[u].d[1];
	if (l < u) ls =  build(l,u - 1,d ^ 1);
	if (r > u) rs =  build(u + 1,r,d ^ 1);
	pup(u);
	return u;
}
void qmx(int u){
	LL t = equal(u) ? -INF : (abs(e[u].d[0] - x) + abs(e[u].d[1] - y)),d[2];
	if (ls) d[0] = getdx(ls); else d[0] = -INF;
	if (rs) d[1] = getdx(rs); else d[1] = -INF;
	cmax(mx,t); t = d[0] <= d[1];
	if (d[t] > mx) qmx(e[u].s[t]); t ^= 1;
	if (d[t] > mx) qmx(e[u].s[t]);
}
void qmn(int u){
	int t = equal(u) ? INF : (abs(e[u].d[0] - x) + abs(e[u].d[1] - y)),d[2];
	if (ls) d[0] = getd(ls); else d[0] = INF;
	if (rs) d[1] = getd(rs); else d[1] = INF;
	cmin(mn,t); t = d[0] >= d[1];
	if (d[t] < mn) qmn(e[u].s[t]); t ^= 1;
	if (d[t] < mn) qmn(e[u].s[t]);
}
int main(){
	n = read();
	for (int i = 1; i <= n; i++) a[i].d[0] = read(),a[i].d[1] = read();
	rt = build(1,n,0);
	LL ans = INF;
	for (int i = 1; i <= n; i++){
		x = a[i].d[0]; y = a[i].d[1];
		mx = 0; qmx(rt);
		mn = INF; qmn(rt);
		ans = min(ans,mx - mn);
	}
	printf("%lld\\n",ans);
	return 0;
}

BZOJ2648 / BZOJ2716 两道比较卡常数

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define LL long long int
#define max(a,b) (a > b ? a : b)
#define min(a,b) (a < b ? a : b)
#define cmax(a,b) (a < b ? a = b : a)
#define cmin(a,b) (a > b ? a = b : a)
#define ls e[u].s[0]
#define rs e[u].s[1]
using namespace std;
const int maxn = 1000005,maxm = 500005,INF = 1000000000;
inline int read(){
	int out = 0,flag = 1; char c = getchar();
	while (c < 48 || c > 57){if (c == \'-\') flag = -1; c = getchar();}
	while (c >= 48 && c <= 57){out = (out << 3) + (out << 1) + c - 48; c = getchar();}
	return out * flag;
}
struct node{
	int d[2],s[2],x[2],y[2];
}e[maxn];
int m,n,x,y,D,dd[2],ans,rt;
struct point{int d[2];}p[maxm];
inline bool cmp(const point& a,const point& b){
	return a.d[D] < b.d[D];
}
int getd(int u){
	int tmp = 0;
	tmp += max(e[u].x[0] - x,0);
	tmp += max(x - e[u].x[1],0);
	tmp += max(e[u].y[0] - y,0);
	tmp += max(y - e[u].y[1],0);
	return tmp;
}
inline void pup(int u){
	if (ls){
        cmin(e[u].x[0],e[ls].x[0]); cmax(e[u].x[1],e[ls].x[1]);
        cmin(e[u].y[0],e[ls].y[0]); cmax(e[u].y[1],e[ls].y[1]);
    }
    if (rs){
        cmin(e[u].x[0],e[rs].x[0]); cmax(e[u].x[1],e[rs].x[1]);
        cmin(e[u].y[0],e[rs].y[0]); cmax(e[u].y[1],e[rs].y[1]);
    }
}
int build(int l,int r,int d){
	D = d; int u = l + r >> 1;
	nth_element(p + l,p + u,p + r + 1,cmp);
	e[u].d[0] = e[u].x[0] = e[u].x[1] = p[u].d[0];
	e[u].d[1] = e[u].y[0] = e[u].y[1] = p[u].d[1];
	if (l < u) ls = build(l,u - 1,d ^ 1);
	if (r > u) rs = build(u + 1,r,d ^ 1);
	pup(u);
	return u;
}
void insert(int u,int d){
	if (e[u].d[d] < dd[d]){
		if (ls) insert(ls,d ^ 1);
		else {
			ls = ++n;
			e[ls].d[0] = e[ls].x[0] = e[ls].x[1] = dd[0];
			e[ls].d[1] = e[ls].y[0] = e[ls].y[1] = dd[1];
		}
	}
	else {
		if (rs) insert(rs,d ^ 1);
		else {
			rs = ++n;
			e[rs].d[0] = e[rs].x[0] = e[rs].x[1] = dd[0];
			e[rs].d[1] = e[rs].y[0] = e[rs].y[1] = dd[1];
		}
	}
	pup(u);
}
void query(int u){
	int t = abs(e[u].d[0] - x) + abs(e[u].d[1] - y),d[2];
	if (ls) d[0] = getd(ls); else d[0] = INF;
	if (rs) d[1] = getd(rs); else d[1] = INF;
	cmin(ans,t); t = d[0] >= d[1];
	if (d[t] < ans) query(e[u].s[t]); t ^= 1;
	if (d[t] < ans) query(e[u].s[t]);
}
int main(){
	n = read(); m = read();
	for (int i = 1; i <= n; i++){
		p[i].d[0] = read();
		p[i].d[1] = read();
	}
	rt = build(1,n,0);
	int t;
	for (int i = 1; i <= m; i++){
		t = read(); x = read(); y = read();
		if (t & 1){
			dd[0] = x; dd[1] = y;
			insert(rt,0);
		}
		else {
			ans = INF;
			query(rt);
			printf("%d\\n",ans);
		}
	}
	return 0;
}

以上是关于K-D tree入门更新ing的主要内容,如果未能解决你的问题,请参考以下文章

k-d tree

K-D Tree

简单题(K-D Tree)

k-d tree模板练习

k-d tree学习总结

BZOJ 2648: SJY摆棋子(K-D Tree)