标签:Prim int Highways 算法 POJ mp maxn include
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)
{
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;
}
标签:Prim,int,Highways,算法,POJ,mp,maxn,include 来源: https://www.cnblogs.com/alking1001/p/11600669.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。