POJ_1751_建图+kruskal

POJ 1751

  • 已经建好的边先用并查集指向在一起,然后kruskal
  • 由于是特判,边的输出顺序无关
    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    31
    32
    33
    34
    35
    36
    37
    38
    39
    40
    41
    42
    43
    44
    45
    46
    47
    48
    49
    50
    51
    52
    53
    54
    55
    56
    57
    58
    59
    60
    61
    62
    63
    64
    65
    66
    67
    68
    69
    70
    71
    #include <iostream>
    #include <string.h>
    #include <algorithm>
    #include <math.h>

    using namespace std;
    const int maxn = 1e6;

    struct point{
    int x,y;
    point(){}
    point(int a,int b){
    x = a; y = b;
    }
    }points[maxn];
    struct node {
    int from,to;
    double div;
    node() {}
    node(int a,int b,double c) {
    from = a; to = b; div = c;
    }
    bool operator<(const node& tmp)const {
    return div < tmp.div;
    }
    } e[maxn];
    int n,m;
    int pre[maxn];
    int find(int x) {
    if (x == pre[x]) return x;
    else return pre[x] = find(pre[x]);
    }
    double Distance(point& a,point& b) {
    double dx = (a.x - b.x)*(a.x - b.x);
    double dy = (a.y - b.y)*(a.y - b.y);
    return sqrt(dx + dy);
    }

    int main() {
    // freopen("a.txt","r",stdin);
    scanf("%d",&n);
    for (int i=1; i<=n; i++) pre[i] = i;

    for (int i=1; i<=n; i++) {
    int x,y; scanf("%d%d",&x,&y);
    points[i] = point(x,y);
    }
    int cnt = 0;
    for (int i=1; i<=n; i++) {
    for (int j=i+1; j<=n; j++) {
    double div = Distance(points[i],points[j]);
    e[cnt++] = node(i,j,div);
    // e[cnt++] = node(j,i,div);
    }
    }
    scanf("%d",&m);
    for (int i=1; i<=m; i++) {
    int a,b; scanf("%d%d",&a,&b);
    pre[find(a)] = find(b);
    }
    sort(e,e+cnt);
    for (int i=0; i<cnt; i++) {
    int from = e[i].from; int to = e[i].to;
    int roota = find(from),rootb = find(to);
    if ( roota != rootb ) {
    printf("%d %d\n",from,to);
    pre[roota] = rootb;
    }
    }
    return 0;
    }

本博客所有文章除特别声明外,均采用 CC BY-SA 4.0 协议 ,转载请注明出处!