HDU 1102 Constructing Roads(最小生成树,基础题)

时间:2023-03-09 04:49:53
HDU 1102 Constructing Roads(最小生成树,基础题)

注意标号要减一才为下标,还有已建设的路长可置为0

#define  _CRT_SECURE_NO_WARNINGS

#include <stdio.h>
#include<string.h>
#include <malloc.h>
#include<stdlib.h>
#include<algorithm>
#include<iostream>
using namespace std; #define M 110
#define inf 999999999
int mat[M][M]; int prim(int n,int sta)
{
int mark[M],dis[M];
int i,sum=,j,flag;
for(i=;i<n;i++)
{
dis[i]=mat[sta][i];
mark[i]=;
}
mark[sta]=;
for(i=;i<n;i++)
{
int minn=inf;
for(j=;j<n;j++)
{
if(dis[j]<minn&&mark[j]==)
minn=dis[j],flag=j;
}
mark[flag]=;
sum+=dis[flag];
for(j=;j<n;j++)
{
if(dis[j]>mat[flag][j])
dis[j]=mat[flag][j];
}
}
return sum;
} int main()
{
int n,i,j,a,b,m;
while(scanf("%d",&n)!=EOF)
{
for(i=;i<n;i++)
{
for(j=;j<n;j++)
{
scanf("%d",&mat[i][j]); }
}
scanf("%d",&m);
for(i=;i<m;i++)
{
scanf("%d%d",&a,&b);//原来还是这里的问题,注意下标和标号还是有区别的,要减一
mat[a-][b-]=mat[b-][a-]=;//如果两村之间的路已经连接好,则他们之间的路程设为0
}
printf("%d\n",prim(n,));
}
return ;
}