BZOJ 2648 SJY摆棋子

2648: SJY摆棋子

Time Limit: 20 Sec  Memory Limit: 128 MB
Submit: 5202  Solved: 1815
[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

鸣谢 孙嘉裕

KD-Tree的详解有文章

#include <bits/stdc++.h>
#define ll long long
#define inf 10000010
#define eps 1e-7
using namespace std;
inline int read(){
	int x=0;int f=1;char ch=getchar();
	while(!isdigit(ch)) {if(ch=='-') f=-1;ch=getchar();}
	while(isdigit(ch)) {x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
const int MAXN=1e6+10;
struct node{
	int d[2],mn[2],mx[2],l,r;
}p[MAXN],T[MAXN],t;
int D,n,m,ans,root;
inline bool operator < (node n,node m){
	return n.d[D]<m.d[D];
}
inline int dis(node n,node m){
	return abs(n.d[0]-m.d[0])+abs(n.d[1]-m.d[1]);
}
namespace KDTree{
	inline void update(int root){
		for(int i=0;i<=1;i++){
			if(T[root].l) T[root].mn[i]=min(T[T[root].l].mn[i],T[root].mn[i]),T[root].mx[i]=max(T[T[root].l].mx[i],T[root].mx[i]);
			if(T[root].r) T[root].mn[i]=min(T[T[root].r].mn[i],T[root].mn[i]),T[root].mx[i]=max(T[T[root].r].mx[i],T[root].mx[i]);
		}
	}
	inline int build(int l,int r,int now){
		D=now;
		int mid=(l+r)>>1;
		nth_element(p+l,p+mid,p+r+1);
		T[mid]=p[mid];
		for(int i=0;i<2;i++){
			T[mid].mn[i]=T[mid].mx[i]=T[mid].d[i];
		}
		if(l<mid) T[mid].l=build(l,mid-1,now^1);
		if(r>mid) T[mid].r=build(mid+1,r,now^1);
		update(mid);
		return mid;
	}
	inline int get(int k,node p){
		int tmp=0;
		for(int i=0;i<=1;i++){
			tmp+=max(0,T[k].mn[i]-p.d[i]);
			tmp+=max(0,p.d[i]-T[k].mx[i]);
		}
		return tmp;
	}
	inline void insert(int k,int now){
		if(t.d[now]>=T[k].d[now]){
			if(T[k].r) insert(T[k].r,now^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.d[i];
			}
		}
		else{
			if(T[k].l) insert(T[k].l,now^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.d[i];
			}
		}
		update(k);
	}
	inline void query(int k,int now){
		int d,dl=inf,dr=inf;
		d=dis(T[k],t);
		ans=min(ans,d);
		if(T[k].l) dl=get(T[k].l,t);
		if(T[k].r) dr=get(T[k].r,t);
		if(dl<dr){
			if(dl<ans) query(T[k].l,now^1);
			if(dr<ans) query(T[k].r,now^1);
		}
		else{
			if(dr<ans) query(T[k].r,now^1);
			if(dl<ans) query(T[k].l,now^1);
		}
	}
	void init(){
		n=read();m=read();
		for(int i=1;i<=n;i++){
			p[i].d[0]=read();p[i].d[1]=read();
		}
		root=build(1,n,0);
	}
	inline int query(node p){
		ans=inf;query(root,0);
		return ans;
	}
	void solve(){
		for(int i=1;i<=m;i++){
			int op=read();t.d[0]=read();t.d[1]=read();
			if(op==1) insert(root,0);
			else printf("%d\n",query(t));
		}
	}
}
int main(){
	using namespace KDTree;
	init();
	solve();
	return 0;
}