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

题目

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