51nod 1213 二维曼哈顿距离最小生成树 树状数组

题意

二维平面上有N个坐标为整数的点，点x1 y1同点x2 y2之间的距离为：横纵坐标的差的绝对值之和，即：Abs(x1 - x2) + Abs(y1 - y2)（也称曼哈顿距离）。求这N个点所组成的完全图的最小生成树的边权之和。
n<=50000,0<=xi,yi<=1000000

分析

有个小结论就是以一个点为原点建立直角坐标系，在每45度内只会向距离该点最近的一个点连边。
证明比较繁琐，可以点这里。
然后就可以用数据结构来帮助建图了。

代码

#include<iostream>#include<cstdio>#include<cstdlib>#include<cstring>#include<algorithm>using namespace std;typedef long long LL;const int N=50005;const int inf=(int)1e9;int n,cnt,c[N],d[N],w[N],f[N],p[N];struct edge{int x,y,w;}e[N*10];struct data{int x,y,id,val;}a[N];int read(){    int x=0,f=1;char ch=getchar();    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}    return x*f;}int find(int x){    if (f[x]==x) return x;    else return f[x]=find(f[x]);}bool cmp(data a,data b){    return a.x<b.x||a.x==b.x&&a.y<b.y;}bool cmp1(edge a,edge b){    return a.w<b.w;}void ins(int x,int y,int z){    while (x<=n)    {        if (y<c[x]) c[x]=y,d[x]=z;        x+=x&(-x);    }}int query(int x){    int ans=inf,pos=0;    while (x)    {        if (c[x]<ans) ans=c[x],pos=d[x];        x-=x&(-x);    }    return pos;}int main(){    n=read();    for (int i=1;i<=n;i++) a[i].x=read(),a[i].y=read(),a[i].id=i,w[i]=a[i].x;    sort(w+1,w+n+1);    int w1=unique(w+1,w+n+1)-w-1;    for (int i=1;i<=n;i++) p[i]=lower_bound(w+1,w+w1+1,a[i].x)-w;    for (int i=1;i<=n;i++) w[i]=a[i].y;    sort(w+1,w+n+1);    w1=unique(w+1,w+n+1)-w-1;    for (int dir=0;dir<4;dir++)    {        if (dir==1||dir==3) for (int i=1;i<=n;i++) swap(a[i].x,a[i].y);        else if (dir==2) for (int i=1;i<=n;i++) a[i].x=-a[i].x;        for (int i=1;i<=n;i++) a[i].val=a[i].y-a[i].x,w[i]=a[i].val;        sort(a+1,a+n+1,cmp);sort(w+1,w+n+1);        int w1=unique(w+1,w+n+1)-w-1;        for (int i=1;i<=n;i++) c[i]=inf,d[i]=0;        for (int i=n;i>=1;i--)        {            a[i].val=lower_bound(w+1,w+w1+1,a[i].val)-w;            int x=query(n-a[i].val+1);            if (x) e[++cnt].x=a[i].id,e[cnt].y=a[x].id,e[cnt].w=abs(a[i].x-a[x].x)+abs(a[i].y-a[x].y);            ins(n-a[i].val+1,a[i].x+a[i].y,i);        }    }    LL ans=0;    sort(e+1,e+cnt+1,cmp1);    for (int i=1;i<=n;i++) f[i]=i;    for (int i=1;i<=cnt;i++)    {        int x=find(e[i].x),y=find(e[i].y);        if (x!=y) f[x]=y,ans+=e[i].w;    }    printf("%lld",ans);    return 0;}