POJ 3241 Object Clustering (曼哈顿距离最小生成树)

题意： 询问平面上的点的曼哈顿距离最小生成树第n-k小边的长度，点数为10000。

解法：对于曼哈顿距离最小生成树，朴素的解法需要建n^2条边进行kruskal，显然要TLE。2010莫涛论文中提到了一个可以将边优化到4*n条的方法，其实就是利用了一个性质：以任意一个点为端点将平面分成均匀45度角的8个块，那么在生成树的最优解中，每个块与这个点至多有1条边，即一个点最多向8个方向最近的点各连接一条边。这个证明比较简单，自己想比较有意思。

论文内容相当详细了，此处略去300字。

/* Created Time: Monday, November 04, 2013 PM04:10:01 CST */
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
const int INF = 0x3f3f3f3f;
const int N = 2016;
const int M = 10010;
struct point {
    int x,y,id;
    void read() { scanf("%d%d",&x,&y); }
    bool operator < (const point &tt) const {
        return x<tt.x || x==tt.x && y<tt.y;
    }
}dot[M];
struct E {
    int u,v,w;
    E () {}
    E (int _u,int _v,int _w) :
        u(_u),v(_v),w(_w) {}
    bool operator < (const E &tt) const {
        return w<tt.w;
    }
}g[M<<2];
int n,k,C[N],D[N],tot,fa[M];
int Find(int x) { return fa[x]==x ? x : fa[x] = Find(fa[x]); }
bool merge(int a,int b) {
    a = Find(a),b = Find(b);
    if (a==b) return false;
    fa[a] = b;
    n -- ;
    return true;
}
void query(int val,int w,int cid) {
    int id = -1,minx = INF;
    val += 1000;
    for (int i = val; i < N; i += i&-i) 
        if (C[i]<minx) {
            minx = C[i];
            id = D[i];
        }
    if (id!=-1)
        g[tot++] = E(cid,id,minx-w);
}
void update(int val,int w,int cid) {
    val += 1000;
    for (int i = val; i > 0; i -= i&-i) 
        if (w<C[i]) {
            C[i] = w;
            D[i] = cid;
        }
}
void go() {
    // 排序，从右往左处理,寻找y-x～N范围内具有最小y+x值的点的id，这里用的树状数组
    sort(dot,dot+n);
    for (int i = 0; i < N; i ++) 
        C[i] = INF;
    for (int i = n-1; i >= 0; i --) {
        query(dot[i].y-dot[i].x,dot[i].y+dot[i].x,dot[i].id);
        update(dot[i].y-dot[i].x,dot[i].y+dot[i].x,dot[i].id);
    }
}
int work() {
    if (k==n) return 0;

    // 处理4个方向的边
    // (x,y)
    go();
    // (y,x)
    for (int i = 0; i < n; i ++) 
        swap(dot[i].x,dot[i].y);
    go();
    // (y,-x)
    for (int i = 0; i < n; i ++)
        dot[i].y = -dot[i].y;
    go();
    // (-x,y)
    for (int i = 0; i < n; i ++)
        swap(dot[i].x,dot[i].y);
    go();

    // Kruskal
    sort(g,g+tot);
    for (int i = 0; i < n; i ++)
        fa[i] = i;
    for (int i = 0; i < tot; i ++)
        if (merge(g[i].u,g[i].v) && n==k)
            return g[i].w;
    return -1;
}
int main() {
    scanf("%d%d",&n,&k);
    for (int i = 0; i < n; i ++) {
        dot[i].read();
        dot[i].id = i;
    }
    printf("%d\n",work());
    return 0;
}