hiho刷题日记——第二十六天最小生成树一·Prim算法

题目

思路

代码

#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;
}