[曼哈顿距离最小生成树 模板题] POJ 3241 Object Clustering

我不会啊
老老实实去学 曼哈顿距离最小生成树

以一个点为原点建立直角坐标系,在每45度内只会向距离该点最近的一个点连边
这样只会有 O(n) 条边
怎么求最近点 是个二维偏序 排完序后 树状数组维护即可

#include
#include
#include
#include
using namespace std;
typedef pair<int,int> abcd;

inline char nc(){
  static char buf[100000],*p1=buf,*p2=buf;
  return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline void read(int &x){
  char c=nc(),b=1;
  for (;!(c>='0' && c<='9');c=nc()) if (c=='-') b=-1;
  for (x=0;c>='0' && c<='9';x=x*10+c-'0',c=nc()); x*=b;
}

const int N=10005;

struct edge{
  int u,v,w;
  edge(int u=0,int v=0,int w=0):u(u),v(v),w(w) { }
  bool operator < (const edge &B) const{ return w2]; int tot;

int fat[N];
inline void Init(int n){
  for (int i=1;i<=n;i++) fat[i]=i;
}
inline int Fat(int u){
  return u==fat[u]?u:fat[u]=Fat(fat[u]);
}
inline int Merge(int x,int y){
  x=Fat(x); y=Fat(y); if (x==y) return 0;
  fat[x]=y; return 1;
}

inline int Krus(int n,int K){
  Init(n); sort(ed+1,ed+tot+1);
  for (int i=1;i<=tot;i++)
    if (Merge(ed[i].u,ed[i].v))
      if (!(--K))
    return ed[i].w;
}

namespace BIT{
  abcd c[N]; int maxn;
  inline void init(int n){
    maxn=n; for (int i=1;i<=n;i++) c[i]=abcd(1<<30,0);
  }
  inline void modify(int x,abcd r){
    for (int i=x;i;i-=i&-i)
      c[i]=min(c[i],r);
  }
  inline int query(int x){
    abcd ret(1<<30,0);
    for (int i=x;i<=maxn;i+=i&-i)
      ret=min(ret,c[i]);
    return ret.second;
  }
}

struct P{
  int x,y,idx;
  bool operator < (const P &B)const { return x==B.x?yfriend int dist(P a,P b){ return abs(a.x-b.x)+abs(a.y-b.y); }
}p[N];

int n,K;

int sx[N],icnt;
inline int Bin(int x){
  return lower_bound(sx+1,sx+icnt+1,x)-sx;
}

inline void Build(){
  using namespace BIT;
  for (int d=0;d<4;d++){
    if (d&1) for (int i=1;i<=n;i++) swap(p[i].x,p[i].y);
    if (d==2) for (int i=1;i<=n;i++) p[i].x=-p[i].x;
    sort(p+1,p+n+1);
    icnt=0; for (int i=1;i<=n;i++) sx[++icnt]=p[i].y-p[i].x;
    sort(sx+1,sx+icnt+1); icnt=unique(sx+1,sx+icnt+1)-sx-1;
    init(icnt);
    for (int i=n;i;i--){
      int pos=Bin(p[i].y-p[i].x),v=query(pos);
      if (v)
    ed[++tot]=edge(p[i].idx,p[v].idx,dist(p[i],p[v]));
      modify(pos,abcd(p[i].x+p[i].y,i));
    }
  }
}

int main(){
  freopen("t.in","r",stdin);
  freopen("t.out","w",stdout);
  read(n); read(K);
  for (int i=1;i<=n;i++) read(p[i].x),read(p[i].y),p[i].idx=i;
  Build();
  printf("%d\n",Krus(n,n-K));
  return 0;
}

你可能感兴趣的:(生成树)