[BZOJ5465][APIO2018]选圆圈(KD-Tree)

题意：给你n个圆，每次选择半径最大的，将它和与它相交的圆全部删去，输出每个圆是在哪次被删的。

KD树模板题。用一个矩形框住这个圆，就可以直接剪枝了。为了防止被卡可以将点旋转一个角度，为了保险还可以多转几个角度。

 #include<cmath>
 #include<cstdio>
 #include<algorithm>
 #define rep(i,l,r) for (int i=(l); i<=(r); i++)
 typedef double db;
 using namespace std;

 const int N=;
 const db inf=1e20,eps=1e-,alpha=acos(-)/;
 int n,rt,ans[N];
 db x,y,r;
 struct P{ db x,y,r; int id; }cur,a[N];
 struct Tr{ int ls,rs; db x1,y1,x2,y2; P c; }T[N];

 bool cmpx(const P &a,const P &b){ return a.x<b.x; }
 bool cmpy(const P &a,const P &b){ return a.y<b.y; }
 bool cmpr(const P &a,const P &b){ return (a.r==b.r) ? a.id<b.id : a.r>b.r; }

 void F(int x,int y){
     T[x].x1=min(T[x].x1,T[y].x1); T[x].x2=max(T[x].x2,T[y].x2);
     T[x].y1=min(T[x].y1,T[y].y1); T[x].y2=max(T[x].y2,T[y].y2);
 }

 void upd(int x){
     if (ans[T[x].c.id]) T[x].x1=T[x].y1=inf,T[x].x2=T[x].y2=-inf;
     else{
         T[x].x1=T[x].c.x-T[x].c.r; T[x].x2=T[x].c.x+T[x].c.r;
         T[x].y1=T[x].c.y-T[x].c.r; T[x].y2=T[x].c.y+T[x].c.r;
     }
     if (T[x].ls) F(x,T[x].ls);
     if (T[x].rs) F(x,T[x].rs);
 }

 int build(int l,int r,int k){
     if (l>r) return ;
     int mid=(l+r)>>; nth_element(a+l,a+mid,a+r+,k?cmpy:cmpx);
     T[mid].c=a[mid];
     T[mid].ls=build(l,mid-,k^); T[mid].rs=build(mid+,r,k^);
     upd(mid); return mid;
 }

 db sqr(db x){ return x*x; }
 bool Out(int x){ return (T[x].x2<cur.x-cur.r-eps) || (T[x].x1>cur.x+cur.r+eps) || (T[x].y2<cur.y-cur.r-eps) || (T[x].y1>cur.y+cur.r+eps); }
 bool chk(P &a){ return sqr(a.x-cur.x)+sqr(a.y-cur.y)<=sqr(a.r+cur.r)+eps; }

 void que(int x){
     if (Out(x)) return;
     if (!ans[T[x].c.id] && chk(T[x].c)) ans[T[x].c.id]=cur.id;
     if (T[x].ls) que(T[x].ls);
     if (T[x].rs) que(T[x].rs);
 }

 int main(){
     freopen("apiob.in","r",stdin);
     freopen("apiob.out","w",stdout);
     scanf("%d",&n);
     rep(i,,n)
         scanf("%lf%lf%lf",&x,&y,&r),a[i]=(P){x*cos(alpha)+y*sin(alpha),y*cos(alpha)-x*sin(alpha),r,i};
     rt=build(,n,); sort(a+,a+n+,cmpr);
     rep(i,,n) if (!ans[a[i].id]) ans[a[i].id]=a[i].id,cur=a[i],que(rt);
     rep(i,,n) printf("%d ",ans[i]); puts("");
     return ;
 }