洛谷 P1546 最短網路 Agri-Net(最小生成樹_Prim)
阿新 • • 發佈:2019-02-16
最小生成樹模板,大家都說是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);
}