HDU1875

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

tyg_zx

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

阅读量664

点赞数

分类专栏：最小生成树

本文链接：https://blog.csdn.net/qq_15681523/article/details/38788639

版权

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

4 篇文章 0 订阅

订阅专栏

还是最小生成树的模板，在模板上限制一下条件就可以了

#include<iostream>
#include<string.h>
#include<cmath>
#include<iomanip>
using namespace std;
#define MAX  0x3f3f3f3f; 
int number;
double map[101][101];
int visit[101];
double low[101];
double prim()
{
    memset(visit,0,sizeof(visit));//刚开始就是没初始化WA了3次，哎，自己呀
    int i,j,pos;
    double result=0;
    visit[0]=1;
    pos=0;
    for(i=0;i<number;i++)
        low[i]=MAX;
    for(i=0;i<number;i++)
    {
        if(i!=pos&&map[pos][i]>=10&&map[pos][i]<=1000)
            low[i]=map[pos][i];
    }
    for(i=0;i<number-1;i++)
    {
        double min=MAX;
        for(j=0;j<number;j++)
        {
            if(visit[j]==0&&min>low[j]&&low[j]<=1000&&low[j]>=10)
            {
                min=low[j];
                pos=j;
            }
        }
        result+=min;
        visit[pos]=1;
        for(j=0;j<number;j++)
        {
            if(visit[j]==0&&low[j]>map[pos][j]&&map[pos][j]>=10&&map[pos][j]<=1000)
                low[j]=map[pos][j];
        }
    }
    return result;
    
}
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        double  a[101],b[101],c,d;
        int i,j;
        cin>>number;
        for(i=0;i<number;i++)
        {
            for(j=0;j<number;j++)
                map[i][j]=MAX;
        }       
        i=0;
        double m;
        for(i=0;i<number;i++)
        {
            cin>>c>>d;
            a[i]=c;
            b[i]=d;
        }
        for(i=0;i<number;i++)
        {
            for(j=i+1;j<number;j++)
            {
                m=sqrt(fabs((a[i]-a[j])*(a[i]-a[j])+(b[i]-b[j])*(b[i]-b[j])));
                //cout<<m<<" ";
                map[i][j]=m;
                map[j][i]=m;
            }
        }
        double ans=prim();
        int flag=1;
        for(i=0;i<number;i++)
        {
            if(visit[i]==0)
            {
                flag=0;
                break;
            }
        }
        if(flag==1)
            cout<<setiosflags(ios::fixed)<<setprecision(1)<<100*ans<<endl;
        else
            cout<<"oh!"<<endl;
    }
}