POJ1751 PRIM基础题（模板）

最新推荐文章于 2019-06-10 17:34:47 发布

原创最新推荐文章于 2019-06-10 17:34:47 发布 · 491 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#mina #struct #ini #c

本文介绍了一种使用Prim算法来寻找最小生成树(MST)的方法，并通过C++代码实现了该算法。作者首先分享了一个存在效率问题的自写MST函数，随后展示了优化后的Prim算法模板。该算法适用于带权重的无向图，通过不断选择最小边来构建连接所有顶点的子图。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

那个mst函数是我自己写的。有点超时。

学习了一下别人的prim模板。

以后要改进了。

#include<iostream>

#include<cmath>
using namespace std;
int n,m;
const int N=755;
const int M=1005;
const int inf=99999999;
struct
{
int x,y;
}town[N];
double mat[N][N];
bool visit[N];
void mst()
{
visit[1]=true;
for(int count=1;count<n;count++)
{
int min=inf;
int mina=0,minb=0;
for(int i=1;i<=n;i++)
{
if(visit[i])
{
for(int j=1;j<=n;j++)
{
if(!visit[j])
{
if(min>mat[i][j])
min=mat[i][j],mina=i,minb=j;
}
}
}
}
visit[minb]=true;
if(min!=0)
printf("%d %d\n",mina,minb);
}
}
double key[N];
int pre[N];
void prim()
{
int k;
for(int i=1;i<=n;i++)
key[i]=mat[1][i],pre[i]=1;
visit[1]=true;
for(int i=1;i<n;i++)
{
int min=inf;
for(int j=1;j<=n;j++)
{
if(!visit[j]&&min>key[j])
{
min=key[j];
k=j;
}
}
visit[k]=true;
if(mat[k][pre[k]]!=0)
printf("%d %d\n",k,pre[k]);
for(int j=1;j<=n;j++)
{
if(!visit[j]&&key[j]>mat[k][j])
pre[j]=k,key[j]=mat[k][j];
}
}
}
int main()
{
//freopen("C:\\Users\\wuyanyisb\\Desktop\\1.txt","r",stdin);
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d%d",&town[i].x,&town[i].y);
}
scanf("%d",&m);
int a,b;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
mat[i][j]=inf;
}
for(int i=1;i<=m;i++)
{
scanf("%d%d",&a,&b);
mat[a][b]=0;
mat[b][a]=0;
}
for(int i=1;i<=n;i++)
for(int j=i+1;j<=n;j++)
{
if(mat[i][j]!=0)
{
int tmp=(town[i].x-town[j].x)*(town[i].x-town[j].x)+(town[i].y-town[j].y)*(town[i].y-town[j].y);
mat[i][j]=tmp;
mat[j][i]=mat[i][j];
}
}
//mst();
prim();
return 0;
}