prim算法模版 hdu 1875

来源：互联网发布：南沙人工智能产业园编辑：程序博客网时间：2024/05/20 14:27
#include <stdio.h>#include <string.h>#include <math.h>#define N 110#define MAX 1000#define INF 0x3f3f3f3fint n;int x[N],y[N];//保存每个岛的坐标 double g[N][N];//保存岛与岛之间的距离 double dis(int i , int j){return sqrt( 1.0*((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j])) );}void input(){    int i,j;    scanf("%d",&n);    for(i=1; i<=n; i++)    scanf("%d%d",&x[i],&y[i]);//小岛i的坐标     for(i=1; i<=n; i++)    for(j=1 ;j<=n; j++)    {g[i][j]=g[j][i]=dis(i,j);//求出岛屿i j之间的距离     if(g[i][j] < 10 || g[i][j] > 1000)//不符合要求的直接设为无穷大 g[i][j] = g[j][i] = INF;}return ;}void prim(){    double dis[N],min,sum;    int vis[N];    int v,i,j,k;memset(vis, 0, sizeof(vis)); //初始化岛屿全为未在点集合内     for(i=1; i<=n; i++)dis[i]=g[1][i];//记录i点到集合1的距离     dis[1]=0;//本身到本身的距离为0vis[1] = 1; //标记1已在集合内     for(v=1; v<n; v++)  //还要纳入n-1个点    {        min=INF; k=1;        for(i=1; i<=n; i++)        if(!vis[i] && dis[i] < min)//如果i点未在集合内  并且小于最小值         {min = dis[i];//记录最小值 k = i;//记录编号 }if(min == INF){printf("oh!\n");return ;}        vis[k] = 1;//标记为已在集合内         for(i=1; i<=n; i++)//更新其余点到集合1的距离         if(!vis[i] && dis[i] > g[k][i])//如果i未在集合1内并且 当前保存的i到集合的距离大于新更新的值         dis[i]=g[k][i];//更新最小值     }    sum = 0;for(i = 2; i <= n; ++i)sum += dis[i] * 100.0;printf("%.1lf\n", sum);    return ;}int main(){    int T;    scanf("%d",&T);    while(T--)    {        input();        prim();    }    return 0;}
0 0