BZOJ2648 SJY摆棋子

传送门

题解:
既然Hint上明确指出“kdtree可以过”,那么这就是一道kdtree板题了。
继续存板子。
总之在这里就是 在第0层以x为关键字 第1层以y为关键字 第2层又以x为关键字 以此类推。
然后查找就相当于优化了一下爆搜(…)答案更有可能在哪边就先往哪边找。

代码:

#include
#include
#define maxn 1000005
#define INF 1000000000
using namespace std;
int D,n,m,root;
struct data
{
	int c[2],mn[2],mx[2],l,r;
	data(int x=0,int y=0) { l=r=0,c[0]=x,c[1]=y; }
	bool operator < (const data &rhs) const { return c[D]<rhs.c[D]; }
}p[maxn];
struct kdtree
{
	int ans;
	data t[maxn*2],T;
	void Pushup(int k)
	{
		data l=t[t[k].l],r=t[t[k].r];
		for(int i=0;i<=1;i++)
		{
			if(t[k].l)
			{
				t[k].mn[i]=min(t[k].mn[i],l.mn[i]);
				t[k].mx[i]=max(t[k].mx[i],l.mx[i]);
			}
			if(t[k].r)
			{
				t[k].mn[i]=min(t[k].mn[i],r.mn[i]);
				t[k].mx[i]=max(t[k].mx[i],r.mx[i]);
			}
		}
	}
	int Build(int l,int r,int dim)
	{
		D=dim;
		int mid=(l+r)>>1;
		nth_element(p+l,p+mid,p+r+1);
		t[mid]=p[mid];
		for(int i=0;i<=1;i++) t[mid].mn[i]=t[mid].mx[i]=t[mid].c[i];
		if(l<mid) t[mid].l=Build(l,mid-1,dim^1);
		if(r>mid) t[mid].r=Build(mid+1,r,dim^1);
		Pushup(mid);
		return mid;
	}
	void Insert(int k,int dim)
	{
		if(T.c[dim]>=t[k].c[dim])
		{
			if(t[k].r) Insert(t[k].r,dim^1);
			else
			{
				t[k].r=++n; t[n]=T;
				for(int i=0;i<=1;i++) t[n].mn[i]=t[n].mx[i]=t[n].c[i];
			}
		}
		else
		{
			if(t[k].l) Insert(t[k].l,dim^1);
			else
			{
				t[k].l=++n; t[n]=T;
				for(int i=0;i<=1;i++) t[n].mn[i]=t[n].mx[i]=t[n].c[i];
			}
		}
		Pushup(k);
	}
	void Insert(data p) { T=p; Insert(root,0); }
	int dis(data p,data q) { return abs(p.c[0]-q.c[0])+abs(p.c[1]-q.c[1]); }
	int get(data p)
	{
		int sum=0;
		for(int i=0;i<=1;i++) sum+=max(p.mn[i]-T.c[i],0)+max(T.c[i]-p.mx[i],0);
		return sum;
	}
	void Query(int k)
	{
		int d=dis(t[k],T),dl=INF,dr=INF;
		ans=min(ans,d);
		if(t[k].l) dl=get(t[t[k].l]);
		if(t[k].r) dr=get(t[t[k].r]);
		if(dl<dr)
		{
			if(dl<ans) Query(t[k].l);
			if(dr<ans) Query(t[k].r);
		}
		else
		{
			if(dr<ans) Query(t[k].r);
			if(dl<ans) Query(t[k].l);
		}
	}
	int Query(data p) { ans=INF,T=p; Query(root); return ans; }
}kd;
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%d%d",&p[i].c[0],&p[i].c[1]);
	root=kd.Build(1,n,0);
	while(m--)
	{
		int tp,x,y;
		scanf("%d%d%d",&tp,&x,&y);
		if(tp==1) kd.Insert(data(x,y));
		else printf("%d\n",kd.Query(data(x,y)));
	}
}

你可能感兴趣的:(懵逼了半天终于AC,kdtree)