luoguP4169 [Violet]天使玩偶/SJY摆棋子 K-Dtree

P4169 [Violet]天使玩偶/SJY摆棋子

链接

luogu

思路

luogu以前用CDQ一直过不去。

bzoj还是卡时过去的。

今天终于用k-dtree给过了。

代码

#include <bits/stdc++.h>
using namespace std;
const int INF=0x3f3f3f3f,N=1e6+7;;
const double alph(0.75);
int WD,ans,rub[N],tot,top;
struct Point {
int x[2];
bool operator < (const Point b) const {
return x[WD]<b.x[WD];
}
}p[N];
struct node {
int mi[2],ma[2],ls,rs,siz;
Point tp;
}e[N];
int newnode() {
if(top) return rub[top--];
return ++tot;
}
void up(int u) {
for(int i=0;i<2;++i) {
e[u].mi[i]=e[u].ma[i]=e[u].tp.x[i];
if(e[u].ls) {
e[u].mi[i]=min(e[u].mi[i],e[e[u].ls].mi[i]);
e[u].ma[i]=max(e[u].ma[i],e[e[u].ls].ma[i]);
}
if(e[u].rs) {
e[u].mi[i]=min(e[u].mi[i],e[e[u].rs].mi[i]);
e[u].ma[i]=max(e[u].ma[i],e[e[u].rs].ma[i]);
}
}
e[u].siz=e[e[u].ls].siz+e[e[u].rs].siz+1;
}
int build(int l,int r,int wd) {
if(l>r) return 0;
int u=newnode(),mid=(l+r)>>1;
WD=wd;
nth_element(p+l,p+mid,p+r+1);
e[u].tp=p[mid];
e[u].ls=build(l,mid-1,wd^1);
e[u].rs=build(mid+1,r,wd^1);
up(u);
return u;
}
void pia(int u,int num) {
if(e[u].ls) pia(e[u].ls,num);
p[num+e[e[u].ls].siz+1]=e[u].tp;
rub[++top]=u;
if(e[u].rs) pia(e[u].rs,num+e[e[u].ls].siz+1);
}
void check(int &u,int wd) {
if(alph*e[u].siz<e[e[u].ls].siz||alph*e[u].siz<e[e[u].rs].siz)
pia(u,0),u=build(1,e[u].siz,wd);
}
void insert(Point tmp,int &u,int wd) {
if(!u) {u=newnode(),e[u].tp=tmp;e[u].ls=e[u].rs=0;up(u);return;}
if(e[u].tp.x[wd]<tmp.x[wd]) insert(tmp,e[u].rs,wd^1);
else insert(tmp,e[u].ls,wd^1);
up(u),check(u,wd);
}
int getdis(Point tmp,int u) {
int res=0;
for(int i=0;i<=1;++i)
res+=max(0,e[u].mi[i]-tmp.x[i])+max(0,tmp.x[i]-e[u].ma[i]);
return res;
}
int dis(Point a,Point b) {
return abs(a.x[0]-b.x[0])+abs(a.x[1]-b.x[1]);
}
void query(Point tmp,int u) {
ans=min(ans,dis(e[u].tp,tmp));
int dl=INF,dr=INF;
if(e[u].ls) dl=getdis(tmp,e[u].ls);
if(e[u].rs) dr=getdis(tmp,e[u].rs);
if(dl<dr) {
if(dl<ans) query(tmp,e[u].ls);
if(dr<ans) query(tmp,e[u].rs);
} else {
if(dr<ans) query(tmp,e[u].rs);
if(dl<ans) query(tmp,e[u].ls);
}
}
int main() {
int opt,n,m,rt;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;++i) scanf("%d%d",&p[i].x[0],&p[i].x[1]);
rt=build(1,n,0);
while(m--) {
Point tmp;
scanf("%d%d%d",&opt,&tmp.x[0],&tmp.x[1]);
if(opt==1) insert(tmp,rt,0);
else ans=INF,query(tmp,rt),printf("%d\n",ans);
}
return 0;
}
上一篇:(转)一文学会用 Tensorflow 搭建神经网络


下一篇:从零开始用 Flask 搭建一个网站(二)