洛谷4631 [APIO2018] Circle selection 选圆圈 （KD树）

qwq纪念AC450

一开始想这个题想复杂了。

首先，正解的做法是比较麻烦的。
qwqq
那么就不如来一点暴力的东西，看到平面上点的距离的题，不难想到(KD-Tree)

我们用类似平面最近点对那个题一样的维护方式，对于一个子树内部，分别维护每一个维度的最大值和最小值，还有半径的最大值。

然后(sort)一遍，从半径大到小依次(query)，每次(query)的时候，对于当前点，合法的条件是他和目标点的距离要小于等于两个圆的半径的和。

那么对于子树的估价函数，我们默认如果当前目标点的当前维度的范围是(mn-mx)之间的话，距离就是(0)，否则取一个最短距离，然后和上述要求一样的比较方式。进行剪枝。

这样就能直接通过洛谷的数据了，但是由于(loj)的数据比较强，所以还需要将原来的点绕着一个点旋转一下。

那么假设我们旋转的角度是(phi)

那么我们需要将原来那两个坐标乘上一个矩阵

cos -sin
sin cos

直接上代码了

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<queue>
#include<cmath>
#include<map>
#include<set>
#define pb push_back
#define mk make_pair
#define ll long long
#define lson ch[x][0]
#define rson ch[x][1]
#define double long double
#define int long long

using namespace std;

inline int read()
{
  int x=0,f=1;char ch=getchar();
  while (!isdigit(ch)) {if (ch=='-') f=-1;ch=getchar();}
  while (isdigit(ch)) {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
  return x*f;
}

const int maxn = 4e5+1e2;
const double phi = 1.04719755116;
const double eps = 1e-4;

struct KD{
	double d[2],mx[2],mn[2];
	double bj,mr;
	int l,r;
	int num;
};

KD now,t[maxn];
int root,ymh,n,m;
int ans[maxn];

struct Node{
  double x,y;
  double r;
  int num;
};

Node a[maxn];

bool cmp(Node a,Node b)
{
	if (a.r==b.r) return a.num<b.num;
	return a.r>b.r;
}

bool operator <(KD a,KD b)
{
	return a.d[ymh]<b.d[ymh];
}

void up(int root)
{
	for (int i=0;i<=1;i++)
	{
		if (t[root].l) t[root].mn[i]=min(t[root].mn[i],t[t[root].l].mn[i]);
		if (t[root].l) t[root].mx[i]=max(t[root].mx[i],t[t[root].l].mx[i]);
		if (t[root].l) t[root].mr=max(t[root].mr,t[t[root].l].mr);
		if (t[root].r) t[root].mr=max(t[root].mr,t[t[root].r].mr);
		if (t[root].r) t[root].mn[i]=min(t[root].mn[i],t[t[root].r].mn[i]);
		if (t[root].r) t[root].mx[i]=max(t[root].mx[i],t[t[root].r].mx[i]);
	}
}

void build(int &x,int l,int r,int dd)
{
	ymh = dd;
	int mid = l+r >> 1;
	x=mid;
	nth_element(t+l,t+mid,t+r+1);
	for (int i=0;i<=1;i++) t[x].mn[i]=t[x].mx[i]=t[x].d[i];
	t[x].mr=t[x].bj;
	if (l<x) build(t[x].l,l,mid-1,dd^1);
	if (x<r) build(t[x].r,mid+1,r,dd^1);
	up(x);
}

inline double getdis(int x)
{
	double ans=0;
	for (int i=0;i<=1;i++) ans=ans+(t[x].d[i]-now.d[i])*(t[x].d[i]-now.d[i]);
	return ans;
}

inline double calc(int x)
{
	double ans=0;
	for (int i=0;i<=1;i++) 
	{
	  double tmp=0;
	  if (now.d[i]>=t[x].mn[i] && now.d[i]<=t[x].mx[i]) tmp=0;
	  else tmp=min((t[x].mn[i]-now.d[i])*(t[x].mn[i]-now.d[i]),(t[x].mx[i]-now.d[i])*(t[x].mx[i]-now.d[i]));
	  ans=ans+tmp;
    }
	return ans;
}

void query(int x)
{
	//cout<<now.num<<endl;
	//cout<<x<<" "<<t[x].l<<" "<<t[x].r<<" "<<t[x].mr<<" "<<t[x].d[1]<<endl;;
	if (!x) return;
	double d = getdis(x);
	double d1 = calc(t[x].l);
	double d2 = calc(t[x].r);
	//cout<<d1<<" "<<(t[t[x].l].mr+now.bj)<<" "<<d2<<endl;
	if (!ans[t[x].num] && (t[x].bj+now.bj)*(t[x].bj+now.bj)-d>=-eps) ans[t[x].num]=now.num;
	if ((t[t[x].l].mr+now.bj)*(t[t[x].l].mr+now.bj)-d1>=-eps) query(t[x].l);
	if ((t[t[x].r].mr+now.bj)*(t[t[x].r].mr+now.bj)-d2>=-eps) query(t[x].r);
}


signed main()
{
  n=read();
  for (int i=1;i<=n;i++)
  {
  	 double x=read(),y=read(),r=read();
  	 double tmp = x;
	 x=cos(phi)*x-sin(phi)*y;
	 y=sin(phi)*tmp+cos(phi)*y;
	 t[i].num=i;
	 t[i].d[0]=x;
	 t[i].d[1]=y;
	 t[i].bj=r;
	 a[i].x=x;
	 a[i].y=y;
	 a[i].num=i;
	 a[i].r=r; 
	 //printf("%.2lf %.2lf %.2lf
",x,y,r);
	 //cout<<endl;
  }
  build(root,1,n,0);
  //cout<<"********"<<endl;
  sort(a+1,a+1+n,cmp);
  for (int i=1;i<=n;i++)
  {
  	 if (ans[a[i].num]) continue;
  	 now.d[0]=a[i].x;
  	 now.d[1]=a[i].y;
  	 now.bj=a[i].r;
  	 now.num=a[i].num;
  	 query(root);
  	 //out<<"*******"<<endl;
  }
  for (int i=1;i<=n;i++) cout<<ans[i]<<" ";
  return 0;
}