Highways POJ-1751 最小生成树 Prim算法

题意

有一个N个城市M条路的无向图,给你N个城市的坐标,然后现在该无向图已经有M条边了,问你还需要添加总长为多少的边能使得该无向图连通.输出需要添加边的两端点编号即可。

解题思路

这个可以使用最短路里面的Prim算法来实现，对于已经连接的城市，处理方式是令这两个城市之间的距离等于0即可。

prim算法可以实现我们具体的路径输出，Kruskal算法暂时还不大会。

代码实现

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int inf=0x3f3f3f3f;
const int maxn=1e3;
struct Node{
int x, y;
}node[maxn];
int mp[maxn][maxn];
int vis[maxn];
int dis[maxn];//这里的dis存储的是其他各个点，到最小生成树中任意一点的最小值。
int line[maxn];
int n, m, ans=0;
void init()
{
for(int i=1; i<=n; i++)
{
dis[i]=mp[i][1];//开始的时候任选1号顶点加入到生成树中。这里是其他点到1号顶点的距离。
line[i]=1; //默认没有加入到生成树的点距离生成树中最近距离的点是1；
vis[i]=0; //默认没有点加入到生成树中。
}
}
bool prim()
{
    ans=0;
vis[1]=1;
for(int i=1; i<n; i++)
{
int tmp=inf, k;
for(int j=1; j<=n; j++)
{
if(!vis[j] && dis[j]<tmp)
{
tmp=dis[j];
k=j;
}
}
if(tmp==inf) return false;//生成最小树失败，该图不是连通的。
vis[k]=1;
        ans+=dis[k];
if(mp[k][line[k]]!=0)//这里判断不为0，是因为有的路已经修好了，就不用输出了
{
printf("%d %d\n", k, line[k]); //输出需要链接的两个点
}
for(int j=1; j<=n; j++) //以新加入生成树的点作为中间点，看看能优化
{
if(!vis[j] && dis[j] > mp[j][k])
{
line[j]=k;
dis[j]=mp[j][k];
}
}
}
    return true;
}
int main()
{
while(scanf("%d", &n)!=EOF)
{
for(int i=1; i<=n; i++)
{
scanf("%d%d", &node[i].x, &node[i].y);
}
for(int i=1; i<=n; i++)
{
for(int j=1; j<=n; j++)
{
int tmp=(node[i].x-node[j].x)*(node[i].x-node[j].x)+(node[i].y-node[j].y)*(node[i].y-node[j].y);
mp[i][j]=tmp;
mp[j][i]=tmp;
}
}
scanf("%d", &m);
int x, y;
for(int i=1; i<=m; i++)
{
scanf("%d%d", &x, &y);
mp[x][y]=mp[y][x]=0;
}
init();
prim();
}
return  0;
}