HDU 1162 prime+邻接矩阵
prime+邻接矩阵模板
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <vector>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <string>
#include <iostream>
#include <algorithm>
using namespace std;
#define MAXN 110
struct Node//定义点
{
double x,y;
} node[MAXN];
double cost[MAXN][MAXN];//耗费矩阵
double distance(Node a,Node b)//两点的距离
{
return sqrt((a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y));
}
//***********************************************************************
//套用prim模板
//***********************************************************************
#define typec double
const double INF=0x3f3f3f3f;
int vis[MAXN];//已经找过的点
double lowc[MAXN];//未遍历集合中的点到已遍历的最近距离
double prim(double cost[][MAXN],int n)//点从0~n-1
{
int i,j,p;
double minc,res=0;
memset(vis,0,sizeof(vis));
vis[0]=1;
for(i=1; i<n; i++) lowc[i]=cost[0][i];
for(i=1; i<n; i++)//最小树上一共有n-1条边
{
minc=INF;//最小距离
p=-1;//记录最小点位
for(j=0; j<n; j++)
if(vis[j]==0&&minc>lowc[j])
{
minc=lowc[j];
p=j;
}
if(minc==INF)return -1;//原图不连通
res+=minc;
vis[p]=1;
for(j=0; j<n; j++)//只要更新这个点对应的相接点的距离就行
if(vis[j]==0&&lowc[j]>cost[p][j])
lowc[j]=cost[p][j];
}
return res;
}
//********************************************************************
int main()
{
int i,j,n;
while(scanf("%d",&n)!=EOF)
{
for(i=0; i<n; i++)
scanf("%lf%lf",&node[i].x,&node[i].y);
for(i=0; i<n; i++)
{
for(j=0; j<n; j++)
{
if(i==j)cost[i][j]=0;
else
cost[i][j]=distance(node[i],node[j]);
}
}
printf("%.2lf\n",prim(cost,n));
}
return 0;
}