uva 10397 - Connect the Campus(最小生成树&并查集）

 

题目大意：给出n,然后再给出n个点的坐标，要连接着n个坐标，使得总距离最小，但是有m对点已经连接，输入m，和m组a和b表示a和b两点已经连接。

 

解题思路：kruskal算法，每次将最短的边拿出来考虑，判断该边的两个端点是否联通，如果未联通，则加入；否则跳过。

 

 

#include <stdio.h>  
#include <string.h>  
#include <math.h>  
#include <algorithm>  
using namespace std;  
const int N = 760;;  
const int M = 600000;  
  
struct state {  
    double dis;  
    int l;  
    int r;  
}s[M];;  
  
int n, cnt, f[N], x[N], y[N];  
  
int getfather(int c) {  
    return c == f[c] ? c : f[c] = getfather(f[c]);  
}  
  
double add(int i, int j) {  
    s[cnt].dis = sqrt((x[i] - x[j]) * (x[i] - x[j]) + (y[i] - y[j]) * ( y[i] - y[j]));  
    s[cnt].l = i;  
    s[cnt].r = j;  
    cnt++;  
}  
  
void init() {  
    cnt = 0;  
    for (int i = 0; i < n; i++) {  
        f[i] = i; // Init f;  
        scanf("%d%d", &x[i], &y[i]);  
        for (int j = 0; j < i; j++)  
            add(i, j);  
    }  
  
    int m, a, b;  
    scanf("%d", &m);  
    for (int i = 0; i < m; i++) {  
        scanf("%d%d", &a, &b);  
        int p = getfather(a - 1), q = getfather(b - 1);  
        if (p != q)  
            f[q] = p;  
    }  
}  
  
bool cmp(const state& a, const state& b) {  
    return b.dis - a.dis > 1e-9;  
}  
  
double kruskal() {  
    sort(s, s + cnt, cmp);;  
    double sum = 0;  
    for (int i = 0; i < cnt; i++) {  
        int p = getfather(s[i].l), q = getfather(s[i].r);  
        if (p != q) {  
            f[q] = p;  
            sum += s[i].dis;  
        }  
    }  
    return sum;  
}  
  
int main () {  
    while (scanf("%d", &n) == 1) {  
        init();  
        printf("%.2lf\n", kruskal());  
    }  
    return 0;  
}  

 

 

补充：软件开发 , C++ ,