Range Search (kD Tree) 范围搜索

链接：https://vjudge.net/problem/Aizu-DSL_2_C
思想：通过构造二叉树，将x与y在树上了交替进行排序（如第一层对x排序，第二层对y进行排序，第三层再对x，以此类推），可以将原本为o（n方）的复杂度降低为o（n（logn方））,注意数据量大用scanf和printf替代cin和cout!

代码：
#include
#include
#include
using namespace std;

class Node{//树结点的构造
public:
int location,l,r,p;//左右父节点
Node(){}
};

class Point{//记录原来的点的序列
public:
int id,x,y;
Point(){}
Point(int id,int x,int y):id(id),x(x),y(y){}

bool operator < (const Point &p)const{
return id }

void print(){
printf("%d\n", id);
}
};

static const int NIL = -1;
static const int MAX =1000010;
int N;
Point P[MAX];
Node T[MAX];
int np;

bool lessX(const Point &p1,const Point &p2){return p1.x bool lessY(const Point &p1,const Point &p2){return p1.y

int maketree(int l,int r,int depth){//递归构造二叉树
if(l>=r)return NIL;
int mid = (l+r)/2;
int t = np++;

if(depth%2==0)//按层给x和y交替排序构成二叉树
sort(P+l,P+r,lessX);//按照x排序
else sort(P+l,P+r,lessY);//按照y排序

T[t].location = mid;
T[t].l = maketree(l,mid,depth+1);
T[t].r = maketree(mid+1,r,depth+1);

return t;
}

void find(int v,int sx,int tx,int sy,int ty,int depth,vector &ans){//寻找范围内的点
int x = P[T[v].location].x;
int y = P[T[v].location].y;

if(sx<=x&&x<=tx&&sy<=y&&y<=ty)
ans.push_back(P[T[v].location]);

if(depth%2==0){//按x和y分层的构造方式,采取对应的分层查找方法
if(T[v].l!=NIL&&x>=sx)
find(T[v].l,sx,tx,sy,ty,depth+1,ans);
if(T[v].r!=NIL&&x<=tx)
find(T[v].r,sx,tx,sy,ty,depth+1,ans);
}
else{
if(T[v].l!=NIL&&y>=sy)
find(T[v].l,sx,tx,sy,ty,depth+1,ans);
if(T[v].r!=NIL&&y<=ty)
find(T[v].r,sx,tx,sy,ty,depth+1,ans);
}
}

int main(){
int x,y;
scanf("%d", &N);
for(int i=0;i scanf("%d %d", &x, &y);
P[i] = Point(i,x,y);
T[i].l = T[i].r = T[i].p = NIL;
}

np = 0;
int root = maketree(0,N,0);
//printf("%d%d\n",root,P[T[root].location]);
int n;
scanf("%d", &n);
int sx,tx,sy,ty;
vector result;

for(int j=0;j scanf("%d %d %d %d",&sx,&tx,&sy,&ty);
result.clear();
find(root,sx,tx,sy,ty,0,result);
sort(result.begin(),result.end());
for(int i=0;i result[i].print();
printf("\n");
}
return 0;
}