HDU 2966 In case of failure

In case of failure

http://acm.hdu.edu.cn/showproblem.php?pid=2966

题意：

　　　求平面上距离每个点最近的点。输出平方即可。

分析：

　　k-d tree模板题。k-d树的最近邻搜索。

　　关于k-d tree 划分平面：应该按照维度的方差的大小来划分，取最大的一维，由于这道题只有两维，而且对于每一层先按x，在按y的方案也是可以过的。还有可以按照维度最长的划分。

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;

const int N = 100010;
const int DIM = 2; // dimension
const LL INF = 1e18;

struct Point{
    LL x[2];
}ori[N],p[N],Q; // original point & point
int Cur;

#define Root 1,n,1
#define lson l,m-1,D+1
#define rson m+1,r,D+1

bool cmp(Point a,Point b) {
    return a.x[Cur] < b.x[Cur];
}
LL sqr(LL x) {
    return x * x;
}
LL dist(Point a,Point b) {
    LL res = 0;
    for (int i=0; i<DIM; ++i) 
        res += sqr(a.x[i] - b.x[i]);
    return res ? res : INF;
}
void build(int l,int r,int D) {
    if (l >= r) return ;
    int m = (l + r) >> 1;
    Cur = D & 1;
    nth_element(p+l, p+m, p+r+1, cmp);
    build(lson);
    build(rson);
}
LL query(int l,int r,int D) {
    int cur = D & 1;
    if (l >= r) {
        if (l == r)  return dist(Q, p[l]);
        else return INF;
    }
    int m = (l + r ) >> 1;
    LL res = dist(Q, p[m]),tmp;
    if (Q.x[cur] < p[m].x[cur]) {
        tmp = query(lson);
        if (tmp > sqr(Q.x[cur] - p[m].x[cur])) tmp = min(tmp, query(rson));
    }
    else {
        tmp = query(rson);
        if (tmp > sqr(Q.x[cur] - p[m].x[cur])) tmp = min(tmp, query(lson));
    }
    return min(res, tmp);
}
void solve() {
    int n; scanf("%d",&n); 
    for (int i=1; i<=n; ++i) {
        for (int j=0; j<DIM; ++j) 
            scanf("%lld",&ori[i].x[j]);
        p[i] = ori[i];
    }
    build(Root);
    for (int i=1; i<=n; ++i) {
        Q = ori[i];
        printf("%lld
", query(Root));
    }
}
int main() {
    int Case;scanf("%d",&Case);
    while (Case--) solve();
    return 0;
}