bzoj 2648: SJY摆棋子 （KD-tree）

2648: SJY摆棋子

Time Limit: 20 Sec   Memory Limit: 128 MB
Submit: 3152   Solved: 1078
[ Submit][ Status][ Discuss]

Description

这天，SJY显得无聊。在家自己玩。在一个棋盘上，有N个黑色棋子。他每次要么放到棋盘上一个黑色棋子，要么放上一个白色棋子，如果是白色棋子，他会找出距离这个白色棋子最近的黑色棋子。此处的距离是 曼哈顿距离 即(|x1-x2|+|y1-y2|) 。现在给出N<=500000个初始棋子。和M<=500000个操作。对于每个白色棋子，输出距离这个白色棋子最近的黑色棋子的距离。同一个格子可能有多个棋子。

 

Input

第一行两个数 N M

以后M行，每行3个数 t x y

如果t=1 那么放下一个黑色棋子

如果t=2 那么放下一个白色棋子

Output

对于每个T=2 输出一个最小距离

 

Sample Input

2 3
1 1
2 3
2 1 2
1 3 3
2 4 2

Sample Output

1
2

HINT

 

kdtree可以过

Source

鸣谢 孙嘉裕

[ Submit][ Status][ Discuss]

题解：KD-tree

KD-tree模板题

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define N 1000003
#define inf 2139062143
using namespace std;
struct data
{
	int d[2],mx[2],mn[2],l,r;
}tr[N];
int n,m,cmpd,x,y,opt,ans,root;
void update(int now)
{
	int l=tr[now].l; int r=tr[now].r;
	if (l) {
		tr[now].mx[0]=max(tr[now].mx[0],tr[l].mx[0]);
		tr[now].mx[1]=max(tr[now].mx[1],tr[l].mx[1]);
		tr[now].mn[1]=min(tr[now].mn[1],tr[l].mn[1]);
		tr[now].mn[0]=min(tr[now].mn[0],tr[l].mn[0]);
	}
	if (r) {
		tr[now].mx[0]=max(tr[now].mx[0],tr[r].mx[0]);
		tr[now].mx[1]=max(tr[now].mx[1],tr[r].mx[1]);
		tr[now].mn[1]=min(tr[now].mn[1],tr[r].mn[1]);
		tr[now].mn[0]=min(tr[now].mn[0],tr[r].mn[0]);
	}
}
int cmp(data a,data b)
{
	return a.d[cmpd]<b.d[cmpd]||a.d[cmpd]==b.d[cmpd]&&a.d[!cmpd]<b.d[!cmpd];
}
int build(int l,int r,int d)
{
	cmpd=d;
	int mid=(l+r)/2;
	nth_element(tr+l,tr+mid,tr+r+1,cmp);
	tr[mid].mx[0]=tr[mid].mn[0]=tr[mid].d[0];
	tr[mid].mx[1]=tr[mid].mn[1]=tr[mid].d[1];
	if (l!=mid) tr[mid].l=build(l,mid-1,d^1);
	if (r!=mid) tr[mid].r=build(mid+1,r,d^1);
	update(mid);
	return mid;
}
void insert(int now)
{
	int p=root; int d=0;
	while (true) {
		tr[p].mx[0]=max(tr[p].mx[0],tr[now].mx[0]);
		tr[p].mx[1]=max(tr[p].mx[1],tr[now].mx[1]);
		tr[p].mn[1]=min(tr[p].mn[1],tr[now].mn[1]);
		tr[p].mn[0]=min(tr[p].mn[0],tr[now].mn[0]);
		if (tr[now].d[d]>=tr[p].d[d])
		{
		 if (!tr[p].r) {
		 	tr[p].r=now; return;
		 }else p=tr[p].r;
	    }
	    else {
	    	if (!tr[p].l) {
	    		tr[p].l=now; return;
			}
			else p=tr[p].l;
		}
		d^=1;
	}
}
int dist(int now,int x,int y)
{
	int dis=0;
	if (x<tr[now].mn[0]) dis+=tr[now].mn[0]-x;
	if (x>tr[now].mx[0]) dis+=x-tr[now].mx[0];
	if (y<tr[now].mn[1]) dis+=tr[now].mn[1]-y;
	if (y>tr[now].mx[1]) dis+=y-tr[now].mx[1];
	return dis;
}
void query(int now)
{
	int dl,dr,d0;
	d0=abs(tr[now].d[0]-x)+abs(tr[now].d[1]-y);
	ans=min(ans,d0);
	if (tr[now].l) dl=dist(tr[now].l,x,y);
	else dl=inf;
	if (tr[now].r) dr=dist(tr[now].r,x,y);
	else dr=inf;
	if (dl<dr) {
		if (dl<ans) query(tr[now].l);
		if (dr<ans) query(tr[now].r);
	}
	else {
		if (dr<ans) query(tr[now].r);
		if (dl<ans) query(tr[now].l);
	}
}
int main()
{
	freopen("a.in","r",stdin);
	//freopen("my.out","w",stdout);
	scanf("%d%d",&n,&m);
	for (int i=1;i<=n;i++) scanf("%d%d",&tr[i].d[0],&tr[i].d[1]);
	root=build(1,n,0);
	for (int i=1;i<=m;i++) {
		scanf("%d%d%d",&opt,&x,&y);
		if (opt==1) {
			++n;
			tr[n].mx[0]=tr[n].mn[0]=tr[n].d[0]=x;
	        tr[n].mx[1]=tr[n].mn[1]=tr[n].d[1]=y;
	        insert(n);
		}
		else {
			ans=inf;
			query(root);
			printf("%d\n",ans);
		}
	}
}