1. 程式人生 > >洛谷 P1546 最短網路 Agri-Net(最小生成樹_Prim)

洛谷 P1546 最短網路 Agri-Net(最小生成樹_Prim)

傳送門

最小生成樹模板,大家都說是Kruskal,但brz大神說是稠密圖要用Prim。
由於大神很強我聽大神的

關於Prim演算法和Kruskal看這裡,我覺得他寫得很好

Code:

#include<cstdio>
#include<cstdlib>
#include<cstring>

int n,ans=0;
int a[110][110],d[110];
bool v[110];

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        for(int j=1
;j<=n;j++) scanf("%d",&a[i][j]); memset(d,63,sizeof(d)); for(int i=1;i<=n;i++) d[i]=a[1][i]; memset(v,false,sizeof(v));v[1]=true; for(int i=1;i<=n;i++) { int k=0; for(int j=1;j<=n;j++) if(!v[j] && d[k]>d[j]) k=j; v[k]=true
; for(int j=1;j<=n;j++) if(!v[j] && a[k][j]<d[j]) d[j]=a[k][j]; } for(int i=1;i<=n;i++) ans+=d[i]; printf("%d",ans); }