hdu5992 Finding Hotels 2016青岛区域赛

http://acm.hdu.edu.cn/showproblem.php?pid=5992

按照x,y两维建k-d树，价格<=的时候再记录答案就行了。

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

const int maxl=2e5+10;

int n,m,idx,ans,ansdis;
struct node
{
	int x[3],id;
	bool operator < (const node &b)const
	{
		return x[idx]<b.x[idx];
	}
}a[maxl],b[maxl];
typedef pair<double,node> pp;
struct kdtree
{
	node p[maxl<<2];
	int son[maxl<<2];
	void build(int l,int r,int u,int dep)
	{
		if(l>r)
			return;
		son[u]=r-l+1;
		son[u<<1]=son[u<<1|1]=0;
		idx=dep%2;
		int mid=(l+r)>>1;
		nth_element(b+l,b+mid,b+r+1);
		p[u]=b[mid];
		build(l,mid-1,u<<1,dep+1);
		build(mid+1,r,u<<1|1,dep+1);
	}
	void query(node d,int u,int dep)
	{
		pp nd(0,p[u]);
		for(int i=0;i<2;i++)
			nd.first+=pow(d.x[i]-p[u].x[i],2);
		int id=dep%2,flag=0;
		int x=u<<1,y=u<<1|1;
		if(d.x[id]>=p[u].x[id]) swap(x,y);
		if(son[x])
			query(d,x,dep+1);
		if(!ans)
		{
			if(p[u].x[2]<=d.x[2])
				ans=p[u].id,ansdis=nd.first;
			flag=1;
		}
		else
		{
			if(nd.first==ansdis && p[u].x[2]<=d.x[2])
				ans=min(ans,p[u].id);
			if(nd.first<ansdis && p[u].x[2]<=d.x[2])
				ans=p[u].id,ansdis=nd.first;
			if(pow(p[u].x[id]-d.x[id],2)<ansdis)
				flag=1;
		}
		if(son[y] && flag)
			query(d,y,dep+1);
	}
}kd;

inline void prework()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		for(int j=0;j<3;j++)
		{
			scanf("%d",&a[i].x[j]);
			b[i].x[j]=a[i].x[j];
		}
		a[i].id=b[i].id=i;
	}
	kd.build(1,n,1,0);
}

inline void mainwork()
{
	node d;
	for(int i=1;i<=m;i++)
	{
		for(int j=0;j<3;j++)
			scanf("%d",&d.x[j]);
		ans=0;
		kd.query(d,1,0);
		printf("%d %d %d\n",a[ans].x[0],a[ans].x[1],a[ans].x[2]);
	}
}

int main()
{
	int t;
	scanf("%d",&t);
	for(int i=1;i<=t;i++)
	{
		prework();
		mainwork();
	}
	return 0;
}