K-D Tree
感觉K-D Tree 是将数据组织成易于搜索的结构
剩下的就是暴力搜索和剪枝了
P3769 [CH弱省胡策R2]TATT
先排序一维(目的是当前的点只能从前面的点转移,详见cmp1)
让后按排序完的顺序询问,插入
剪枝:1.当前询问子树的左子树(右子树同)的最长长度maxl<now(当前答案),不进入左子树(右子树)搜索
2.当前子树的左子树(右子树)某个维度i的最小值minl[i]>p.x[i](当前点在此维度的值),不进入左(右)子树
3.对于当前子树,左右 子树maxl谁大先搜谁(这样如果跟新答案,maxl小的另一个子树就符合剪枝1,不用搜了)
#include<bits/stdc++.h>
using namespace std;
const int maxn=5e4+5;
const double alphy=0.725;
int n,top,cnt,now,T,rt,ans,Dim;
struct node{
int x[4],len;
}p[maxn],q[maxn];
struct Tree{
int size,lc,rc,maxl;
int minl[4];
node p;
}tr[maxn];
int rub[maxn];
int newnode(){
if(top)return rub[top--];
return ++T;
}
void update(int k){
tr[k].size=tr[tr[k].lc].size+tr[tr[k].rc].size+1;
tr[k].maxl=max(tr[k].p.len,max(tr[tr[k].lc].maxl,tr[tr[k].rc].maxl));
for(int i=0;i<3;i++){
tr[k].minl[i]=tr[k].p.x[i];
if(tr[k].lc)tr[k].minl[i]=min(tr[k].minl[i],tr[tr[k].lc].minl[i]);
if(tr[k].rc)tr[k].minl[i]=min(tr[k].minl[i],tr[tr[k].rc].minl[i]);
}
}
bool cmp(node a,node b){
return a.x[Dim]<b.x[Dim];
}
int build(int l,int r,int dim)
{
if(l>r)return 0;
int mid=(l+r)>>1,k=newnode();
Dim=dim;
nth_element(p+l,p+mid,p+r+1,cmp);
tr[k].p=p[mid];
tr[k].lc=build(l,mid-1,(dim+1)%3);
tr[k].rc=build(mid+1,r,(dim+1)%3);
update(k);
return k;
}
void work(int k){
if(tr[k].lc)work(tr[k].lc);
if(tr[k].rc)work(tr[k].rc);
rub[++top]=k;
p[++cnt]=tr[k].p;
}
bool bad(int k){return tr[k].size*alphy<tr[tr[k].lc].size||tr[k].size*alphy<tr[tr[k].rc].size;}
void rebuild(int &k,int dim){
cnt=0;
work(k);
k=build(1,cnt,dim);
}
void insert(int &k,node p,int dim)
{
if(!k){
k=newnode();
tr[k].p=p;
tr[k].lc=tr[k].rc=0;
update(k);
return;
}
if(p.x[dim]<=tr[k].p.x[dim])insert(tr[k].lc,p,(dim+1)%3);
else insert(tr[k].rc,p,(dim+1)%3);
update(k);
if(bad(k))rebuild(k,dim);
}
bool is_ok(int k,int x[]){
for(int i=0;i<3;i++){
if(tr[k].p.x[i]>x[i])return 0;
}
return 1;
}
bool is_in(int k,int x[]){
for(int i=0;i<3;i++){
if(tr[k].minl[i]>x[i])return 0;
}
return 1;
}
void query(int k,int x[])
{
int d0=1;
if(is_ok(k,x))d0=tr[k].p.len+1;
now=max(now,d0);
if(tr[tr[k].lc].maxl>tr[tr[k].rc].maxl){
if(tr[k].lc&&tr[tr[k].lc].maxl>=now&&is_in(tr[k].lc,x))query(tr[k].lc,x);
if(tr[k].rc&&tr[tr[k].rc].maxl>=now&&is_in(tr[k].rc,x))query(tr[k].rc,x);
}
else{
if(tr[k].rc&&tr[tr[k].rc].maxl>=now&&is_in(tr[k].rc,x))query(tr[k].rc,x);
if(tr[k].lc&&tr[tr[k].lc].maxl>=now&&is_in(tr[k].lc,x))query(tr[k].lc,x);
}
}
bool cmp1(node a,node b){
if(a.x[3]==b.x[3]){
if(a.x[2]==b.x[2]){
if(a.x[1]==b.x[1]){
return a.x[0]<b.x[0];
}
return a.x[1]<b.x[1];
}
return a.x[2]<b.x[2];
}
return a.x[3]<b.x[3];
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
for(int j=0;j<4;j++)scanf("%d",&q[i].x[j]);
sort(q+1,q+1+n,cmp1);
for(int i=1;i<=n;i++){
now=1; query(rt,q[i].x);
// cout<<now<<endl;
ans=max(ans,now);
q[i].len=now;
insert(rt,q[i],0);
}
cout<<ans<<endl;
return 0;
}
P6247 [SDOI2012]最近最远点对
#include<bits/stdc++.h>
using namespace std;
#define dl double
const int maxn=1e5+5;
const dl inf=1e30;
int n,rt,T,Dim;
struct node{
dl x[2];
}p[maxn];
struct Tree{
int size,lc,rc;
dl maxa[2],mina[2];
node p;
}tr[maxn];
dl now_max=0,now_min=inf,now;
int newnode(){return ++T;}
void update(int k){
for(int i=0;i<=1;i++){
tr[k].mina[i]=tr[k].maxa[i]=tr[k].p.x[i];
if(tr[k].lc){
tr[k].mina[i]=min(tr[k].mina[i],tr[tr[k].lc].mina[i]);
tr[k].maxa[i]=max(tr[k].maxa[i],tr[tr[k].lc].maxa[i]);
}
if(tr[k].rc){
tr[k].mina[i]=min(tr[k].mina[i],tr[tr[k].rc].mina[i]);
tr[k].maxa[i]=max(tr[k].maxa[i],tr[tr[k].rc].maxa[i]);
}
}
}
bool cmp(node a,node b){
return a.x[Dim]<b.x[Dim];
}
int build(int l,int r,int dim)
{
if(l>r)return 0;
int mid=(l+r)>>1,k=newnode();
Dim=dim;
nth_element(p+l,p+mid,p+r+1,cmp);
tr[k].p=p[mid];
tr[k].lc=build(l,mid-1,dim^1);
tr[k].rc=build(mid+1,r,dim^1);
update(k);
return k;
}
dl pow_me(dl x){return x*x;}
dl get_max(int k,dl x,dl y){
dl res=0;
res+=max(pow_me(tr[k].mina[0]-x),pow_me(tr[k].maxa[0]-x));
res+=max(pow_me(tr[k].mina[1]-y),pow_me(tr[k].maxa[1]-y));
return res;
}
void query_max(int k,dl x,dl y){
dl d0=pow_me(tr[k].p.x[0]-x)+pow_me(tr[k].p.x[1]-y),d1,d2;
now=max(now,d0);
if(tr[k].lc)d1=get_max(tr[k].lc,x,y);
if(tr[k].rc)d2=get_max(tr[k].rc,x,y);
if(d1>d2){
if(d1>now)query_max(tr[k].lc,x,y);
if(d2>now)query_max(tr[k].rc,x,y);
}
else{
if(d2>now)query_max(tr[k].rc,x,y);
if(d1>now)query_max(tr[k].lc,x,y);
}
}
dl get_min(int k,dl x,dl y){
dl res=0;
if(x>tr[k].maxa[0])res+=pow_me(tr[k].maxa[0]-x);
else if(x<tr[k].mina[0])res+=pow_me(tr[k].mina[0]-x);
if(y>tr[k].maxa[1])res+=pow_me(tr[k].maxa[1]-y);
else if(y<tr[k].mina[1])res+=pow_me(tr[k].mina[1]-y);
return res;
}
void query_min(int k,dl x,dl y){
dl d0=pow_me(tr[k].p.x[0]-x)+pow_me(tr[k].p.x[1]-y),d1=inf,d2=inf;
if(d0!=0)now=min(d0,now);
if(tr[k].lc)d1=get_min(tr[k].lc,x,y);
if(tr[k].rc)d2=get_min(tr[k].rc,x,y);
if(d1<d2){
if(d1<now)query_min(tr[k].lc,x,y);
if(d2<now)query_min(tr[k].rc,x,y);
}
else{
if(d2<now)query_min(tr[k].rc,x,y);
if(d1<now)query_min(tr[k].lc,x,y);
}
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)scanf("%lf%lf",&p[i].x[0],&p[i].x[1]);
rt=build(1,n,0);
for(int i=1;i<=n;i++)now=0,query_max(rt,p[i].x[0],p[i].x[1]),now_max=max(now_max,now);
for(int i=1;i<=n;i++)now=inf,query_min(rt,p[i].x[0],p[i].x[1]),now_min=min(now_min,now);
// for(int i=1;i<=n;i++)cout<<now_min[i]<<' ';cout<<endl;
printf("%.2lf %.2lf\n",sqrt(now_min),sqrt(now_max));
return 0;
}
P4357 [CQOI2016]K远点对
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=1e5+100;
int n,T,Dim,rt;
struct node{
int x[2];
}p[maxn];
struct Tree{
int lc,rc,size;
int maxa[2],mina[2];
node p;
}tr[maxn];
bool cmp(node a,node b){
return a.x[Dim]<b.x[Dim];
}
int newnode(){return ++T;}
void update(int x){
for(int i=0;i<=1;i++){
tr[x].mina[i]=tr[x].maxa[i]=tr[x].p.x[i];
if(tr[x].lc){
tr[x].mina[i]=min(tr[x].mina[i],tr[tr[x].lc].mina[i]);
tr[x].maxa[i]=max(tr[x].maxa[i],tr[tr[x].lc].maxa[i]);
}
if(tr[x].rc){
tr[x].mina[i]=min(tr[x].mina[i],tr[tr[x].rc].mina[i]);
tr[x].maxa[i]=max(tr[x].maxa[i],tr[tr[x].rc].maxa[i]);
}
}
}
int build(int l,int r,int dim)
{
if(l>r)return 0;
int mid=(l+r)>>1,k=newnode();
Dim=dim;
nth_element(p+l,p+mid,p+r+1,cmp);
tr[k].p=p[mid];
tr[k].lc=build(l,mid-1,dim^1);
tr[k].rc=build(mid+1,r,dim^1);
update(k);
return k;
}
priority_queue<ll,vector<ll> , greater<ll> >q;
ll pow_me(int x){return 1ll*x*x;}
ll get_dis(int k,int x,int y){
ll res=0;
res+=max(pow_me(tr[k].mina[0]-x),pow_me(tr[k].maxa[0]-x));
res+=max(pow_me(tr[k].mina[1]-y),pow_me(tr[k].maxa[1]-y));
return res;
}
void query(int k,int x,int y)
{
ll d0=pow_me(tr[k].p.x[0]-x)+pow_me(tr[k].p.x[1]-y),d1=0,d2=0;
if(tr[k].lc)d1=get_dis(tr[k].lc,x,y);
if(tr[k].rc)d2=get_dis(tr[k].rc,x,y);
if(d0>q.top())q.pop(),q.push(d0);
if(d1>d2){
if(d1>q.top())query(tr[k].lc,x,y);
if(d2>q.top())query(tr[k].rc,x,y);
}
else{
if(d2>q.top())query(tr[k].rc,x,y);
if(d1>q.top())query(tr[k].lc,x,y);
}
}
int main()
{
int K;
scanf("%d%d",&n,&K);
for(int i=1;i<=n;i++)scanf("%d%d",&p[i].x[0],&p[i].x[1]);
rt=build(1,n,0);
for(int i=K*2;i>0;i--)q.push(0);
for(int i=1;i<=n;i++)query(rt,p[i].x[0],p[i].x[1]);
printf("%lld",q.top());
return 0;
}