题意:
求一个图中的最小生成树。
思路:
我们可以用prim算法,可以先从一个点出发,找与它连通的边最短且那个点不在最小生成树里的点,然后加上这条边的值,然后就可以也从这个点出发去找了。
代码:
#include<cstdio>
int n,f[5001][5001],ans,v[5001],mi,t;
int main()
{
scanf("%d",&n);
for (int i=1;i<=n;i++)
for (int j=1;j<=n;j++)
scanf("%d",&f[i][j]);
v[1]=1;
for (int k=1;k<n;k++)
{
mi=2100000000;
for (int i=1;i<=n;i++)
if (v[i])//如果可以从这个点出发
for (int j=1;j<=n;j++)
if (!v[j]&&f[i][j]&&f[i][j]<mi)//如果找的这个点不在生成树里,而且这个点和出发点连通,还有这两个点的边最短
{
mi=f[i][j];//就记录下来
t=j;
}
if (mi!=2100000000)
{
ans+=mi;//把记录的结果加到答案里去,并把刚才找的的点标记起来
v[t]=1;
}
}
printf("%d",ans);
}
更新:
这题还可以用kruskal算法,先让每个点自己成为一个连通分量,然后找一条代价最短的边,且被这条边连接的两点不是一个连通分量的,我们就加上这条最短的边,然后让两个点成为一个相同的连通分量。
代码:
#include<cstdio>
int f[101][101],v[101],ans,n,min1,p,q,t;
int main()
{
scanf("%d",&n);
for (int i=1;i<=n;i++)
for (int j=1;j<=n;j++)
scanf("%d",&f[i][j]);
for (int i=1;i<=n;i++)
v[i]=i;//让每一个点成为一个连通分量
for (int k=1;k<n;k++)
{
min1=2100000000;
for (int i=1;i<=n;i++)
for (int j=1;j<=n;j++)
if (v[i]!=v[j]&&f[i][j]<min1&&f[i][j])//找出一条最短的边,且连接这边的两个点不属于同一连通分量
{
min1=f[i][j];
p=j;
q=i;
}
t=v[p];//找出第二点的连通分量
ans+=min1;//统计答案
for (int j=1;j<=n;j++)
if (v[j]==t) v[j]=v[q];//找到第一个点的连通分量,让它和第二个连通分量的值相等
}
printf("%d",ans);
}