hdu 1875 畅通工程再续

最新推荐文章于 2021-07-27 08:46:09 发布

G_rass

最新推荐文章于 2021-07-27 08:46:09 发布

阅读量363

点赞数

分类专栏：最小生成树

本文链接：https://blog.csdn.net/G_rass/article/details/47790255

版权

最小生成树专栏收录该内容

3 篇文章 0 订阅

订阅专栏

思路：最小生成树，prime算法，这里为了计算方便，距离先存为int型，最后开方时转换为double，

#include <iostream>
#include <cstring>
#include <stdio.h>
#include <cmath>
#define INF 0x3f3f3f;
using namespace std;
int n,m;
//vector <int>g[2000];
int g[2000][2000];
bool vis[2000];
int dis[2000];
struct note
{
    int x,y;
}a[120];
void diss(int i,int j)
{
    int ans=(a[i].x-a[j].x)*(a[i].x-a[j].x)+(a[i].y-a[j].y)*(a[i].y-a[j].y);
    if(ans<100||ans>1000000)
        return ;
    else
        g[i][j]=g[j][i]=ans;
    return ;
}
void prime()
{
    int tmp[120];
    int sum=0;
    for(int i=0;i<n;i++)
        dis[i]=g[0][i];
    vis[0]=true;
    for(int i=1;i<n;i++)
    {
        int ans=INF;
        int pos=-1;
        for(int j=0;j<n;j++)
        {
            if(!vis[j]&&dis[j]<ans)
            {
                ans=dis[j];
                pos=j;
            }
        }
        if(pos==-1)
        {
            printf("oh!\n");
            return ;
        }
        vis[pos]=true;
        tmp[sum++]=dis[pos];
        for(int j=0;j<n;j++)
        {
            if(!vis[j]&&g[pos][j]<dis[j])
                dis[j]=g[pos][j];
        }
    }
    double ans=0;
    for(int i=0;i<sum;i++)
    {
        ans+=sqrt((double)tmp[i]);
    }
    printf("%.1lf\n",ans*100);
    return ; 

}
int main()
{
   // freopen("in.txt","r",stdin);
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        for(int i=0;i<n;i++)
        {
            scanf("%d%d",&a[i].x,&a[i].y);
        }
        memset(g,0x3f,sizeof(g));
        memset(vis,false,sizeof(vis));
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<n;j++)
            {
                if(i==j)
                {
                    g[i][j]=0;
                    continue;
                }
                diss(i,j);
            }
        }
        prime();
    }
    
}