POJ 1258 Agri-Net (prim最小生成树)
浏览数：16 /
时间：2015年06月09日
最小生成树模板题
#include<bits/stdc++.h>
using namespace std;
int n,a;
int dist[120],m[120][120];
void prim()
{
bool p[1020];
for(int i=2;i<=n;i++)
{
p[i]=false;
dist[i]=m[1][i];
}
dist[1]=0,p[1]=true;
for(int i=1;i<=n-1;i++)
{
int min=INT_MAX,k=0;
for(int j=1;j<=n;j++)
{
if(!p[j]&&dist[j]!=0&&dist[j]<min)
{
min=dist[j];
k=j;
}
}
if(k==0)
return;
p[k]=true;
for(int j=1;j<=n;j++)
{
if(!p[j]&&m[k][j]!=0&&(dist[j]==0||dist[j]>m[k][j]))
dist[j]=m[k][j];
}
}
}
int main()
{
while(scanf("%d",&n)==1)
{
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
scanf("%d",&a);
m[i][j]=a;
}
prim();
int sum=0;
for(int i=1;i<=n;i++)
sum+=dist[i];
printf("%d\n",sum);
}
return 0;
}
郑重声明：本站内容如果来自互联网及其他传播媒体，其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享，并不代表本站赞同其观点和对其真实性负责，也不构成任何其他建议。
