POJ 1751 Highways (kruskal)

题目链接：http://poj.org/problem?id=1751

题意是给你n个点的坐标，然后给你m对点是已经相连的，问你还需要连接哪几对点，使这个图为最小生成树。

这里用kruskal不会超时，用prim应该会超时，特别注意在输入的时候不要多组输入，否则会TLE。

 #include <iostream>
 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 #include <vector>
 using namespace std;
 const int MAXN = ;
 struct edge {
     int u , v , cost;
 }a[MAXN * MAXN];
 int x[MAXN] , y[MAXN] , cont , par[MAXN] , cnt , f , ans[MAXN * ];
 bool vis[MAXN][MAXN];

 void init(int n) {
     for(int i =  ; i <= n ; i++) {
         par[i] = i;
     }
     memset(vis , false , sizeof(vis));
     cont = ;
     cnt = n;
 }

 int Find(int n) {
     if(n == par[n])
         return n;
     return (par[n] = Find(par[n]));
 }

 bool cmp(edge a , edge b) {
     return a.cost < b.cost;
 }

 int kruskal() {
     for(int i =  ; i <= cont ; i++) {
         if(cnt == )
             break;
         int u = Find(a[i].u) , v = Find(a[i].v);
         if(u != v) {
             ans[f++] = a[i].u;
             ans[f++] = a[i].v;
             par[u] = v;
             cnt--;
         }
     }
 }

 int main()
 {
     int n , m , u , v;
     scanf("%d" , &n);
     {
         init(n);
         for(int i =  ; i <= n ; i++) {
             scanf("%d %d" , x + i , y + i);
         }
         scanf("%d" , &m);
         while(m--) {
             scanf("%d %d" , &u , &v);
             u = Find(u) , v = Find(v);
             vis[u][v] = vis[v][u] = true;
             if(u != v) {
                 par[u] = v;
                 cnt--;
             }
         }
         for(int i =  ; i <= n ; i++) {
             for(int j =  ; j < i ; j++) {
                 if(vis[i][j])
                     continue;
                 cont++;
                 a[cont].u = j , a[cont].v = i;
                 a[cont].cost = (x[i] - x[j])*(x[i] - x[j]) + (y[i] - y[j])*(y[i] - y[j]);
             }
         }
         sort(a +  , a + cont +  , cmp);
         f = ;
         kruskal();
         for(int i =  ; i < f ; i += ) {
             printf("%d %d\n" , ans[i] , ans[i + ]);
         }
     }
 }