2016HUAS_ACM暑假集训3G - 还是畅通工程

时间:2023-03-09 07:26:43
2016HUAS_ACM暑假集训3G - 还是畅通工程

最小生成树,题目简单。套的Prim模板,其他的题目比较有意义。

Sample Input

3                             //村庄个数
1 2 1                       //连通情况及权值
1 3 2
2 3 4
4
1 2 1
1 3 4
1 4 1
2 3 3
2 4 2
3 4 5
0

Sample Output       //输出最小生成树的权

3

5

 #include<stdio.h>
#include<string.h>
#define size 100
#define inf 0x3f3f3f3f int vellage[size][size],flag[size];
int N,T,a,b,c,i,j,k,ans,maxn; void Init()
{
T=N*(N-)>>;
memset(flag,,sizeof(flag));
memset(vellage,,sizeof(vellage));
} int Prim()
{
ans=;
for(int h=;h<=N;h++)
{
maxn=inf;
for(i=;i<=N;i++)
{
if(flag[i]!=&&vellage[][i]<maxn)
{
maxn=vellage[][i];
k=i;
}
}
ans+=maxn;
flag[k]=;
for(j=;j<=N;j++)
{
if(flag[j]!=&&vellage[k][j]<vellage[][j])
vellage[][j]=vellage[k][j];
}
}
return ans;
} int main()
{
while(scanf("%d",&N),N)
{
Init();
while(T--)
{
scanf("%d %d %d",&a,&b,&c);
vellage[a][b]=vellage[b][a]=c; }
printf("%d\n",Prim());
}
return ;
}