poj 1751 Highways

点击打开链接poj 1751

思路：最小生成树+prime
分析：只要按照prime的思路求出所有的边，然后输出的时候判断是否是已经建好的即可。
注意：可能出现已经把所有的边都建好的情况，这个时候不用输出

代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
using namespace std;
#define MAXN 1000

int n , m;
int vis[MAXN];
int pre[MAXN];
double lowcost[MAXN];
double G[MAXN][MAXN];
struct Point{
    int x;
    int y;
}p[MAXN];
struct Edge{
    int x;
    int y;
}ans[MAXN] , exist[MAXN];

/*初始化边的长度*/
void init(){
    for(int i = 1 ; i <= n ; i++){
       pre[i] = 1;
       for(int j = 1 ; j <= n ; j++){
          double tmp_x = (p[i].x-p[j].x)*(p[i].x-p[j].x);
          double tmp_y = (p[i].y-p[j].y)*(p[i].y-p[j].y);
          G[i][j] = sqrt(tmp_x + tmp_y);
       }
    }
}

/*prime求解最小生成树*/
void prime(){
     memset(vis , 0 , sizeof(vis));
     int pos , cnt;
     vis[1] = 1;
     cnt = 0;
     for(int i = 1 ; i <= n ; i++)
        lowcost[i] = G[1][i];
     for(int i = 1 ; i <= n ; i++){
        pos = -1;
        for(int j = 1 ; j <= n ; j++){
           if(!vis[j] && (pos == -1 || lowcost[j] < lowcost[pos]))
             pos = j;
        }
        if(pos == -1)
            break;
        vis[pos] = 1;
        ans[cnt].x = pre[pos];
        ans[cnt++].y = pos;
        for(int j = 1 ; j <= n ; j++){
           if(!vis[j] && lowcost[j] > G[j][pos]){
              lowcost[j] = G[j][pos];
              pre[j] = pos;
           }
        }
     }
}

/*输出*/
void output(){
     int flag , cnt;
     cnt = 0;
     for(int i = 0 ; i < n-1 ; i++){
         flag = 0;
         for(int j = 0 ; j < m ; j++){
            if(ans[i].x == exist[j].x && ans[i].y == exist[j].y){
               flag = 1;
               break;
            }
            else if(ans[i].x == exist[j].y && ans[i].y == exist[j].x){
               flag = 1;
               break;
            }
         }
         if(!flag){
             cnt++;
             printf("%d %d\n" , ans[i].x , ans[i].y);
         }
     }
}

int main(){
    int begin , end;
    scanf("%d" , &n);
    for(int i = 1 ; i <= n ; i++)
        scanf("%d%d" , &p[i].x , &p[i].y);
    init();
    scanf("%d" , &m);
    for(int i = 0 ; i < m ; i++){
       scanf("%d%d" , &begin , &end);
       G[begin][end] = G[end][begin] = 0;
       exist[i].x = begin;
       exist[i].y = end;
    }
    prime();
    output();
    return 0;
}