题目
用Prim算法做最小生成树
思路
从1开始。将1加入已连接的点。
找到(可以连接到已连接的点)的权值最小的边。
将其的另一端点加入已连接的点。重复到结束。
我这里用了一个d[i]来存储到 i 最小的权值。
代码
#include<cstdio>
#include<cstring>
using namespace std;
const int MAXN=1000+1;
int N,g[MAXN][MAXN],vis[MAXN],d[MAXN],k;
int prim()
{
int ans=0;
memset(vis,0,sizeof(vis));
d[0]=100000;
for(int i=1;i<=N;i++) d[i]=g[1][i];
vis[1]=1;
for(int i=1;i<N;i++)
{
k=0;
for(int j=2;j<=N;j++) if(!vis[j] && d[j]<d[k]) k=j;
vis[k]=1; ans+=d[k];
for(int j=2;j<=N;j++) if(!vis[j] && g[k][j]<d[j]) d[j]=g[k][j];
}
return ans;
}
int main()
{
scanf("%d",&N);
for(int i=1;i<=N;i++)
for(int j=1;j<=N;j++)
scanf("%d",&g[i][j]);
printf("%d",prim());
return 0;
}