POJ 1751 Highways(MST 算法)

题目大意：

给定点的数目n和n个点的坐标，然后给出边的数目m和m组边的信息（通过边两点的序号给出），求在已有边的基础上的最小生成树。输出最小生成树中其他边的信息（边的两端点的信息）。

解题思路：

典型的最小生成树问题，要求输出边的信息，本来用Kruskal算法比较容易标记边的信息，但给出的点是以坐标的形式给出的，所以在点的坐标转化到边的信息方面不太容易，况且没两点之间就要建立一条边。用Prim算法对点的信息存储比较方便（一个二维数组就能搞定），但输出边的信息就有点困难了。我是用Prim算法求最小生成树的，在Prim的基础上我附加了一个数组int father[],用于标记对于第i个点距离最近的点的标号father[i],这两个点之间的边一定在最小生成树里！

代码：

#include<cstdio>
#include<cstring>
using namespace std;
#define V 751
#define typec int // type of cost
const typec inf = 0x3f3f3f3f; // max of cost
int vis[V];
typec lowc[V];
int father[V];  //距离点i最近的点的编号father[i]
int dis[751][751];
int tmpx,tmpy;
int mark[V][V];
struct Pos{
    int x,y;
}cor[800];
Pos line[800];
int th=0;
typec prim(typec cost[][V], int n)  // vertex: 0 ~ n-1
{
    int i, j, p;
    typec minc, res = 0;
    memset(vis, 0, sizeof(vis));
    vis[0] = 1;
    for (i=1; i<n; i++){
        lowc[i] = cost[0][i];
        father[i]=0;
    }
    for (i=1; i<n; i++)
    {
        minc = inf;
        p = -1;
        for (j=0; j<n; j++)
            if (0 == vis[j] && minc > lowc[j])
            {
                minc = lowc[j];
                p = j;
                tmpx=father[j];tmpy=j;
            }
        if (inf == minc) return -1; // 原图不连通
        res += minc;
            line[th].x=tmpx;line[th++].y=tmpy;
        vis[p] = 1;
        for (j=0; j<n; j++)
            if (0 == vis[j] && lowc[j] > cost[p][j]){
                lowc[j] = cost[p][j];
                father[j]=p;   //更新father[]数组
            }
    }
    return res;   //返回最小权值和
}



int main(){
    int n,m;
    int xx,yy;
    while(~scanf("%d",&n)){
        th=0;
        memset(mark,0,sizeof(mark));
        for(int i=0;i<n;i++){
            scanf("%d%d",&cor[i].x,&cor[i].y);
            for(int j=0;j<=i;j++){
                if(i==j){
                    dis[i][j]=0;
                }
                dis[i][j]=dis[j][i]=(cor[i].x-cor[j].x)*(cor[i].x-cor[j].x)+
                (cor[i].y-cor[j].y)*(cor[i].y-cor[j].y);
            }
        }
        scanf("%d",&m);
        for(int i=0;i<m;i++){
            scanf("%d%d",&xx,&yy);
            mark[xx-1][yy-1]=mark[yy-1][xx-1]=1;
            dis[xx-1][yy-1]=dis[yy-1][xx-1]=0;
        }
        int an=prim(dis,n);
        for(int i=0;i<th;i++){
            if(mark[line[i].x][line[i].y]==0)
                printf("%d %d\n",line[i].x+1,line[i].y+1);
        }
    }
    return 0;
}