BZOJ 2648: SJY摆棋子/BZOJ 2716: [Violet 3]天使玩偶 kdtree

2648: SJY摆棋子

Time Limit: 20 Sec  Memory Limit: 128 MB
Submit: 4191  Solved: 1454
[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可以过

kdtree模板题

#include<ctime>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<map>
#include<set>
using namespace std;
typedef double db;
inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	return x*f;
}
const int inf=0X7f7f7f7f;
const int N=1000100;
int n,cur,root;
struct P
{
	int d[2],ls,rs,mn[2],mx[2];
	inline int& operator [](int x){return d[x];}//&&&&&
	friend bool operator <(P x,P y){return x[cur]<y[cur];}//friend
	friend int dis(P x,P y){return abs(x[1]-y[1])+abs(x[0]-y[0]);}//friend
}p[N];
struct kdtree
{
	P tr[N],T;int ans;
	inline void pushup(int k)
	{
		int l=tr[k].ls,r=tr[k].rs;
		for(int i=0;i<2;i++)
		{
			tr[k].mx[i]=tr[k].mn[i]=tr[k][i];
			if(l)
			{tr[k].mn[i]=min(tr[k].mn[i],tr[l].mn[i]);tr[k].mx[i]=max(tr[k].mx[i],tr[l].mx[i]);}
			if(r)
			{tr[k].mn[i]=min(tr[k].mn[i],tr[r].mn[i]);tr[k].mx[i]=max(tr[k].mx[i],tr[r].mx[i]);}
		}
	}
	int build(int l,int r,int now)
	{
		int mid=l+r>>1;cur=now;
		nth_element(p+l,p+mid,p+r+1);
		tr[mid]=p[mid];
		for(int i=0;i<2;i++)tr[mid].mn[i]=tr[mid].mx[i]=tr[mid][i];
		if(l<mid)tr[mid].ls=build(l,mid-1,now^1);
		if(r>mid)tr[mid].rs=build(mid+1,r,now^1);
		pushup(mid);
		return mid;
	}
	void insert(int k,int now)
	{
		if(T[now]>=tr[k][now])
		{
			if(!tr[k].rs)
			{tr[++n]=T;tr[k].rs=n;}
			else insert(tr[k].rs,now^1);
		}
		else
		{
			if(!tr[k].ls)
			{tr[++n]=T;tr[k].ls=n;}
			else insert(tr[k].ls,now^1);
		}
		pushup(k);
	}
	void insert1(int x,int y)
	{
		T[0]=x;T[1]=y;T.ls=T.rs=0;
		for(int i=0;i<2;i++)T.mn[i]=T.mx[i]=T[i];
		insert(root,0);
	}
	inline int getmn(P x)
	{
		int tmp=0;
		for(int i=0;i<2;i++)
		{
			tmp+=max(0,T[i]-x.mx[i]);
			tmp+=max(0,x.mn[i]-T[i]);
		}
		return tmp;
	}
	void query_mn(int k)
	{
		ans=min(ans,dis(tr[k],T));
		int disl=inf,disr=inf;
		if(tr[k].ls)disl=getmn(tr[tr[k].ls]);
		if(tr[k].rs)disr=getmn(tr[tr[k].rs]);
		if(disl<disr)
		{
			if(disl<ans)query_mn(tr[k].ls);
			if(disr<ans)query_mn(tr[k].rs);
		}
		else 
		{
			if(disr<ans)query_mn(tr[k].rs);
			if(disl<ans)query_mn(tr[k].ls);
		}
	}
	void query(int x,int y)
	{
		T[0]=x;T[1]=y;T.ls=T.rs=0;
		ans=inf;query_mn(root);
		printf("%d\n",ans);
	}
}kdtree;
int main()
{
	n=read();int m=read();
	for(int i=1;i<=n;i++)p[i][0]=read(),p[i][1]=read();
	root=kdtree.build(1,n,0);
	while(m--)
	{
		int opt=read(),x=read(),y=read();
		if(opt==1)
		{kdtree.insert1(x,y);}
		else kdtree.query(x,y);
	} 
	return 0;
} 
/*
2 3
1 1
2 3
2 1 2
1 3 3
2 4 2 

1
2
*/