题目链接:UVA - 10034
由于每两个点都可以直接相连,边比较稠密,用prim算法比较好。邻接矩阵存储图。
#include<stdio.h>
#include<string.h>
#include<math.h>
double node[110][2];
double graph[110][110];
double sum;
double lowcost[110];//记录当前集合到各点的最小权值
int nearvex[110];//记录已经加入集合中的点
double dis(double x1,double y1,double x2,double y2)
{
return sqrt((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2));
}
void BuildGraph(int n)
{
int i,j;
for(i=0;i<n;i++)
{
for(j=0;j<n;j++)
{
graph[i][j]=dis(node[i][0],node[i][1],node[j][0],node[j][1]);
}
}
}
void Prim(int n)
{
int i,j,v;
double min;
for(i=0;i<n;i++)
{
lowcost[i]=graph[0][i];
nearvex[i]=0;
}
nearvex[0]=-1;
for(i=1;i<n;i++)
{
min=1<<30;
v=0;
for(j=0;j<n;j++)
{
if(nearvex[j]!=-1&&lowcost[j]<min)
{
min=lowcost[j];
v=j;
}
}
if(v)
{
sum+=lowcost[v];
nearvex[v]=-1;
for(j=0;j<n;j++)
{
if(nearvex[j]!=-1&&graph[v][j]<lowcost[j])
{
lowcost[j]=graph[v][j];
nearvex[j]=v;
}
}
}
}
}
int main()
{
int T,n,i,k;
scanf("%d",&T);
for(k=0;k<T;k++)
{
if(k)printf("\n");
sum=0;
scanf("%d",&n);
for(i=0;i<n;i++)
{
scanf("%lf%lf",&node[i][0],&node[i][1]);
}
BuildGraph(n);
Prim(n);
printf("%.2lf\n",sum);
}
return 0;
}