K-D Tree

K-D Tree

感觉K-D Tree 是将数据组织成易于搜索的结构

剩下的就是暴力搜索和剪枝了

P3769 [CH弱省胡策R2]TATT

先排序一维(目的是当前的点只能从前面的点转移,详见cmp1)

让后按排序完的顺序询问,插入

剪枝:1.当前询问子树的左子树(右子树同)的最长长度maxl<now(当前答案),不进入左子树(右子树)搜索

           2.当前子树的左子树(右子树)某个维度i的最小值minl[i]>p.x[i](当前点在此维度的值),不进入左(右)子树

           3.对于当前子树,左右 子树maxl谁大先搜谁(这样如果跟新答案,maxl小的另一个子树就符合剪枝1,不用搜了)

#include<bits/stdc++.h>
using namespace std;

const int maxn=5e4+5;
const double alphy=0.725;
int n,top,cnt,now,T,rt,ans,Dim;
struct node{
	int x[4],len;
}p[maxn],q[maxn];
struct Tree{
	int size,lc,rc,maxl;
	int minl[4];
	node p;
}tr[maxn];
int rub[maxn];
int newnode(){
	if(top)return rub[top--];
	return ++T;
}
void update(int k){
	tr[k].size=tr[tr[k].lc].size+tr[tr[k].rc].size+1;
	tr[k].maxl=max(tr[k].p.len,max(tr[tr[k].lc].maxl,tr[tr[k].rc].maxl));
	for(int i=0;i<3;i++){
		tr[k].minl[i]=tr[k].p.x[i];
		if(tr[k].lc)tr[k].minl[i]=min(tr[k].minl[i],tr[tr[k].lc].minl[i]);
		if(tr[k].rc)tr[k].minl[i]=min(tr[k].minl[i],tr[tr[k].rc].minl[i]);
	}
}
bool cmp(node a,node b){
	return a.x[Dim]<b.x[Dim];
}
int build(int l,int r,int dim)
{
	if(l>r)return 0;
	int mid=(l+r)>>1,k=newnode();
	Dim=dim;
	nth_element(p+l,p+mid,p+r+1,cmp);
	tr[k].p=p[mid];
	tr[k].lc=build(l,mid-1,(dim+1)%3);
	tr[k].rc=build(mid+1,r,(dim+1)%3);
	update(k);
	return k;
}
void work(int k){
	if(tr[k].lc)work(tr[k].lc);
	if(tr[k].rc)work(tr[k].rc);
	rub[++top]=k;
	p[++cnt]=tr[k].p;
}
bool bad(int k){return tr[k].size*alphy<tr[tr[k].lc].size||tr[k].size*alphy<tr[tr[k].rc].size;}
void rebuild(int &k,int dim){
	cnt=0;
	work(k);
	k=build(1,cnt,dim);
}
void insert(int &k,node p,int dim)
{
	if(!k){
		k=newnode();
		tr[k].p=p;
		tr[k].lc=tr[k].rc=0;
		update(k);
		return;
	}
	if(p.x[dim]<=tr[k].p.x[dim])insert(tr[k].lc,p,(dim+1)%3);
	else insert(tr[k].rc,p,(dim+1)%3);
	update(k);
	if(bad(k))rebuild(k,dim);
}
bool is_ok(int k,int x[]){
	for(int i=0;i<3;i++){
		if(tr[k].p.x[i]>x[i])return 0;
	}
	return 1;
}
bool is_in(int k,int x[]){
	for(int i=0;i<3;i++){
		if(tr[k].minl[i]>x[i])return 0;
	}
	return 1;
}
void query(int k,int x[])
{
	int d0=1;
	if(is_ok(k,x))d0=tr[k].p.len+1;
	now=max(now,d0);
	if(tr[tr[k].lc].maxl>tr[tr[k].rc].maxl){
		if(tr[k].lc&&tr[tr[k].lc].maxl>=now&&is_in(tr[k].lc,x))query(tr[k].lc,x);
		if(tr[k].rc&&tr[tr[k].rc].maxl>=now&&is_in(tr[k].rc,x))query(tr[k].rc,x);
	}
	else{
		if(tr[k].rc&&tr[tr[k].rc].maxl>=now&&is_in(tr[k].rc,x))query(tr[k].rc,x);
		if(tr[k].lc&&tr[tr[k].lc].maxl>=now&&is_in(tr[k].lc,x))query(tr[k].lc,x);
	}
}

bool cmp1(node a,node b){
	if(a.x[3]==b.x[3]){
		if(a.x[2]==b.x[2]){
			if(a.x[1]==b.x[1]){
				return a.x[0]<b.x[0];
			}
			return a.x[1]<b.x[1];
		}
		return a.x[2]<b.x[2];
	}
	return a.x[3]<b.x[3];
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		for(int j=0;j<4;j++)scanf("%d",&q[i].x[j]);
	sort(q+1,q+1+n,cmp1);
	for(int i=1;i<=n;i++){
		now=1; query(rt,q[i].x);
//		cout<<now<<endl;
		ans=max(ans,now);
		q[i].len=now;
		insert(rt,q[i],0);
	}
	cout<<ans<<endl;
	return 0;	
}

P6247 [SDOI2012]最近最远点对

#include<bits/stdc++.h>
using namespace std;

#define dl double
const int maxn=1e5+5;
const dl inf=1e30;
int n,rt,T,Dim;
struct node{
	dl x[2];
}p[maxn];
struct Tree{
	int size,lc,rc;
	dl maxa[2],mina[2];
	node p;
}tr[maxn];
dl now_max=0,now_min=inf,now;
int newnode(){return ++T;}
void update(int k){
	for(int i=0;i<=1;i++){
		tr[k].mina[i]=tr[k].maxa[i]=tr[k].p.x[i];
		if(tr[k].lc){
			tr[k].mina[i]=min(tr[k].mina[i],tr[tr[k].lc].mina[i]);
			tr[k].maxa[i]=max(tr[k].maxa[i],tr[tr[k].lc].maxa[i]);
		}
		if(tr[k].rc){
			tr[k].mina[i]=min(tr[k].mina[i],tr[tr[k].rc].mina[i]);
			tr[k].maxa[i]=max(tr[k].maxa[i],tr[tr[k].rc].maxa[i]);
		}
	}
}
bool cmp(node a,node b){
	return a.x[Dim]<b.x[Dim];
}
int build(int l,int r,int dim)
{
	if(l>r)return 0;
	int mid=(l+r)>>1,k=newnode();
	Dim=dim;
	nth_element(p+l,p+mid,p+r+1,cmp);
	tr[k].p=p[mid];
	tr[k].lc=build(l,mid-1,dim^1);
	tr[k].rc=build(mid+1,r,dim^1);
	update(k);
	return k;
}
dl pow_me(dl x){return x*x;}
dl get_max(int k,dl x,dl y){
	dl res=0;
	res+=max(pow_me(tr[k].mina[0]-x),pow_me(tr[k].maxa[0]-x));
	res+=max(pow_me(tr[k].mina[1]-y),pow_me(tr[k].maxa[1]-y));
	return res;
}
void query_max(int k,dl x,dl y){
	dl d0=pow_me(tr[k].p.x[0]-x)+pow_me(tr[k].p.x[1]-y),d1,d2;
	now=max(now,d0);
	if(tr[k].lc)d1=get_max(tr[k].lc,x,y);
	if(tr[k].rc)d2=get_max(tr[k].rc,x,y);
	if(d1>d2){
		if(d1>now)query_max(tr[k].lc,x,y);
		if(d2>now)query_max(tr[k].rc,x,y);
	}
	else{
		if(d2>now)query_max(tr[k].rc,x,y);
		if(d1>now)query_max(tr[k].lc,x,y);
	}
}
dl get_min(int k,dl x,dl y){
	dl res=0;
	if(x>tr[k].maxa[0])res+=pow_me(tr[k].maxa[0]-x);
	else if(x<tr[k].mina[0])res+=pow_me(tr[k].mina[0]-x);
	if(y>tr[k].maxa[1])res+=pow_me(tr[k].maxa[1]-y);
	else if(y<tr[k].mina[1])res+=pow_me(tr[k].mina[1]-y);
	return res;
}
void query_min(int k,dl x,dl y){
	dl d0=pow_me(tr[k].p.x[0]-x)+pow_me(tr[k].p.x[1]-y),d1=inf,d2=inf;
	if(d0!=0)now=min(d0,now);
	if(tr[k].lc)d1=get_min(tr[k].lc,x,y);
	if(tr[k].rc)d2=get_min(tr[k].rc,x,y);
	if(d1<d2){
		if(d1<now)query_min(tr[k].lc,x,y);
		if(d2<now)query_min(tr[k].rc,x,y);
	}
	else{
		if(d2<now)query_min(tr[k].rc,x,y);
		if(d1<now)query_min(tr[k].lc,x,y);
	}
}

int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)scanf("%lf%lf",&p[i].x[0],&p[i].x[1]);
	rt=build(1,n,0);
	for(int i=1;i<=n;i++)now=0,query_max(rt,p[i].x[0],p[i].x[1]),now_max=max(now_max,now);
	for(int i=1;i<=n;i++)now=inf,query_min(rt,p[i].x[0],p[i].x[1]),now_min=min(now_min,now);
//	for(int i=1;i<=n;i++)cout<<now_min[i]<<' ';cout<<endl;
	printf("%.2lf %.2lf\n",sqrt(now_min),sqrt(now_max));
	return 0;
}

P4357 [CQOI2016]K远点对

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=1e5+100;
int n,T,Dim,rt;
struct node{
	int x[2];
}p[maxn];
struct Tree{
	int lc,rc,size;
	int maxa[2],mina[2];
	node p;
}tr[maxn];

bool cmp(node a,node b){
	return a.x[Dim]<b.x[Dim];
}
int newnode(){return ++T;}
void update(int x){
	for(int i=0;i<=1;i++){
		tr[x].mina[i]=tr[x].maxa[i]=tr[x].p.x[i];
		if(tr[x].lc){
			tr[x].mina[i]=min(tr[x].mina[i],tr[tr[x].lc].mina[i]);
			tr[x].maxa[i]=max(tr[x].maxa[i],tr[tr[x].lc].maxa[i]);
		}
		if(tr[x].rc){
			tr[x].mina[i]=min(tr[x].mina[i],tr[tr[x].rc].mina[i]);
			tr[x].maxa[i]=max(tr[x].maxa[i],tr[tr[x].rc].maxa[i]);
		}
	}
}
int build(int l,int r,int dim)
{
	if(l>r)return 0;
	int mid=(l+r)>>1,k=newnode();
	Dim=dim;
	nth_element(p+l,p+mid,p+r+1,cmp);
	tr[k].p=p[mid];
	tr[k].lc=build(l,mid-1,dim^1);
	tr[k].rc=build(mid+1,r,dim^1);
	update(k);
	return k;
}
priority_queue<ll,vector<ll> , greater<ll> >q; 
ll pow_me(int x){return 1ll*x*x;}
ll get_dis(int k,int x,int y){
	ll res=0;
	res+=max(pow_me(tr[k].mina[0]-x),pow_me(tr[k].maxa[0]-x));
	res+=max(pow_me(tr[k].mina[1]-y),pow_me(tr[k].maxa[1]-y));
	return res;
}
void query(int k,int x,int y)
{
	ll d0=pow_me(tr[k].p.x[0]-x)+pow_me(tr[k].p.x[1]-y),d1=0,d2=0;
	if(tr[k].lc)d1=get_dis(tr[k].lc,x,y);
	if(tr[k].rc)d2=get_dis(tr[k].rc,x,y);
	
	if(d0>q.top())q.pop(),q.push(d0);
	if(d1>d2){
		if(d1>q.top())query(tr[k].lc,x,y);
		if(d2>q.top())query(tr[k].rc,x,y);
	}
	else{
		if(d2>q.top())query(tr[k].rc,x,y);
		if(d1>q.top())query(tr[k].lc,x,y);
	}
}

int main()
{
	int K;
	scanf("%d%d",&n,&K);
	for(int i=1;i<=n;i++)scanf("%d%d",&p[i].x[0],&p[i].x[1]);
	rt=build(1,n,0);
	for(int i=K*2;i>0;i--)q.push(0);
	for(int i=1;i<=n;i++)query(rt,p[i].x[0],p[i].x[1]);
	printf("%lld",q.top());
	return 0;
}

学习于:https://www.cnblogs.com/ShuraEye/p/12069524.html

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
K-D Tree(K-Dimensional Tree)算法是一种基于分治法的数据结构,用于高维空间的搜索和排序。它的基本思想是将多维空间中的点以某种方式分割成更小的子空间,然后在每个子空间中递归地进行搜索。这样可以大大降低搜索的复杂度。 具体来说,K-D Tree算法可以分为以下几步: 1. 选择一个维度,将数据点按照该维度的值进行排序。 2. 找到该维度的中位数,将其作为当前节点,并将数据点分为左右两个子集。 3. 递归地构建左子树和右子树,每次选择一个新的维度进行划分。 4. 最终得到一个K-D Tree。 在搜索时,我们可以从根节点开始,按照一定的规则向下遍历,直到找到目标点或者无法继续向下搜索。具体的规则是: 1. 如果目标点在当前节点的左子树中,则继续向左子树搜索。 2. 如果目标点在当前节点的右子树中,则继续向右子树搜索。 3. 如果目标点和当前节点在选定的维度上的值相等,则说明已经找到目标点。 分治法是一种常见的算法思想,它将一个大规模的问题分解成若干个小规模的子问题,每个子问题独立地求解,然后将这些子问题的解合并起来得到原问题的解。分治法通常包含三个步骤:分解、求解、合并。 具体来说,分治法可以分为以下几步: 1. 分解:将原问题分成若干个子问题,每个子问题规模较小且结构与原问题相同。 2. 求解:递归地求解每个子问题,直到问题规模足够小可以直接求解。 3. 合并:将所有子问题的解合并成原问题的解。 分治法的优点是可以有效地降低算法的时间复杂度。但是它的缺点是需要额外的空间来存储子问题的解,而且分解和合并的过程也需要耗费一定的时间。因此,需要根据实际情况选择合适的算法。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值