来源:HDU1162
水题,解释见代码:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
using namespace std;
int n;
const int MAXN = 105;
const double MAXDOUBLE = 1.0*0x3f3f3f3f;
double x,y;
struct Point{
double x,y;
}p[MAXN];
double edge[MAXN][MAXN];
double tran(Point a,Point b){
return sqrt((a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y));
}
int nxt[MAXN];//集合外各点到我的集合最短距离点的集合
double d[MAXN];//到我已生成点集的最短距离
double prim(){
double ans=0;
for(int i=1;i<=n;i++)
{
d[i]=edge[1][i];
nxt[i]=1;
}
nxt[1]=-1;
int dv=-1;
for(int i=1;i<n;i++){
double dm=MAXDOUBLE;//取的最短距离
dv=-1;//取的最短的点
for(int j=1;j<=n;j++){
if(nxt[j]!=-1&&dm>d[j]){
dv=j;
dm=d[j];
}
}
if(dm==MAXDOUBLE) return -1;
//cout<<dv<<" "<<dm<<endl;
ans+=dm;
nxt[dv]=-1;//已经转移到集合内了,所以要置为-1
for(int j=1;j<=n;j++){
if(nxt[j]!=-1&&d[j]>edge[dv][j]){
d[j]=edge[dv][j];
nxt[j]=dv;
}
}
}
return ans;
}
int main(){
while(scanf("%d",&n)!=EOF){
for(int i=1;i<=n;i++){
scanf("%lf%lf",&p[i].x,&p[i].y);
}
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
if(i==j) edge[i][j]=0;
else
edge[i][j]=tran(p[i],p[j]);
}
}
printf("%.2lf\n",prim());
}
return 0;
}