Bzoj2648:SJY摆棋子:K-D-Tree

最新推荐文章于 2020-01-13 20:23:43 发布

原创最新推荐文章于 2020-01-13 20:23:43 发布 · 624 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#BZOJ #OI #K-D-Tree

OI 同时被 2 个专栏收录

108 篇文章

订阅专栏

K-D-Tree

3 篇文章

订阅专栏

本文介绍了一道名为SJY摆棋子的模板题，并提供了使用K-D树解决该问题的完整代码实现。文章包含核心算法的讲解及代码细节。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：2648:SJY摆棋子

模板题……

~~K-D-Tree早就学过但是并没有写过题所以早就忘了QAQ~~

模板是黄学长的QwQ

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define rep(i,x,y) for (int i=x;i<=y;++i)
using namespace std;
const int maxn=500010;
const int inf=0x7fffffff/2-1;
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
struct point{
	int d[2],mn[2],mx[2],l,r;
	point(int x=0,int y=0){
		l=0,r=0; d[0]=x,d[1]=y;
	}
}p[maxn];
int n,m,D,root;

bool operator < (const point &a,const point &b){
	return a.d[D]<b.d[D];
}

int dis(point a,point b){
	return abs(a.d[0]-b.d[0])+abs(a.d[1]-b.d[1]);
}

struct k_d_tree{
	int ans;
	point t[maxn*2],T;
	void update(int k){
		point l=t[t[k].l],r=t[t[k].r];
		rep(i,0,1){
			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 now){
		D=now;
		int mid=(l+r)>>1;
		nth_element(p+l,p+mid,p+r+1);
		t[mid]=p[mid];
		rep(i,0,1) 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 (mid<r) t[mid].r=build(mid+1,r,now^1);
		update(mid); return mid;
	}
	int getdis(int k,point p){
		int ret=0;
		rep(i,0,1) ret+=max(t[k].mn[i]-p.d[i],0);
		rep(i,0,1) ret+=max(p.d[i]-t[k].mx[i],0);
		return ret;
	}
	void ins(int k,int now){
		if (T.d[now]>=t[k].d[now]){
			if (t[k].r) ins(t[k].r,now^1);
			else{
				t[k].r=++n; t[n]=T;
				rep(i,0,1) t[n].mx[i]=t[n].mn[i]=t[n].d[i];
			}
		}else{
			if (t[k].l) ins(t[k].l,now^1);
			else{
				t[k].l=++n; t[n]=T;
				rep(i,0,1) t[n].mn[i]=t[n].mx[i]=t[n].d[i];
			}
		}update(k);
	}
	void ask(int k,int now){
		int d,dl=inf,dr=inf;
		d=dis(t[k],T); ans=min(ans,d);
		if (t[k].l) dl=getdis(t[k].l,T);
		if (t[k].r) dr=getdis(t[k].r,T);
		if (dl<dr){
			if (dl<ans) ask(t[k].l,now^1);
			if (dr<ans) ask(t[k].r,now^1);
		}else{
			if (dr<ans) ask(t[k].r,now^1);
			if (dl<ans) ask(t[k].l,now^1);
		}
	}
	int query(point p){
		ans=inf; T=p; ask(root,0);
		return ans;
	}
	void insert(point p){
		T=p; ins(root,0);
	}
}kdt;

int main(){
	n=read(); m=read();
	rep(i,1,n) p[i].d[0]=read(),p[i].d[1]=read();
	root=kdt.build(1,n,0);
	rep(i,1,m){
		int opt=read(),x=read(),y=read();
		if (opt==1) kdt.insert(point(x,y));
		else printf("%d\n",kdt.query(point(x,y)));
	}
}