首页 > 代码库 > hdu 1875 畅通工程再续

hdu 1875 畅通工程再续

hdu 1875 畅通工程再续
B - Minimum Spanning Tree 2
Time Limit:1000MS Memory Limit:32768KB 64bit IO Format:%I64d & %I64u
Submit

Status
Description

经过考察小组RPRush对百岛湖的情况充分了解后,决定在符合条件的小岛间建上桥,
所谓符合条件,就是2个小岛之间的距离不能小于10米,也不能大于1000米。
当然,为了节省资金,只要求实现任意2个小岛之间有路通即可。
其中桥的价格为 100元/米。

Input
输入包括多组数据。
输入首先包括一个整数T(T <= 200),代表有T组数据。
每组数据首先是一个整数C(C <= 100),代表小岛的个数,接下来是C组坐标,
代表每个小岛的坐标,这些坐标都是 0 <= x, y <= 1000的整数。

Output
每组输入数据输出一行,代表建桥的最小花费,结果保留一位小数。

如果无法实现工程以达到全部畅通,输出”oh!”.

Sample Input
2
2
10 10
20 20
3
1 1
2 2
1000 1000

Sample Output
1414.2
oh!

 

还是最小生成树 应该还有地方可以优化 200+ms 简直不能忍

#include<stdio.h>#include<string.h>#include<math.h>#include<iostream>#include<algorithm>#include<queue>#include<stack>#define mem(a,b) memset(a,b,sizeof(a))#define ll __int64#define MAXN 1000#define INF 0x7ffffff#define lson l,m,rt<<1#define rson m+1,r,rt<<1|1using namespace std;struct Point {   double x,y;};struct Road{   int f,t;   double w;};Point p[250];Road  r[50000];int fat[250];int sum,cnt,n;double ans;int cmp(Road a,Road b){    return a.w<b.w;}int find(int a){    //return a==fat[a]?a:find(fat[a]);    if(a!=fat[a])    {        fat[a]=find(fat[a]);    }    return fat[a];}void Kruskal(){    for(int i=1;i<=n;i++) fat[i]=i;    for(int i=1;i<cnt;i++)    {        int x=find(r[i].f);        int y=find(r[i].t);        if(x!=y&&r[i].w>=10&&r[i].w<=1000)        {            sum++;            ans+=r[i].w;            fat[y]=x;        }    }    if(sum==n)        {            ans*=100;            printf("%.1f\n",ans);        }        else            cout<<"oh!\n";}int main(){    int i,j,t;    cin>>t;    while(t--)    {        scanf("%d",&n);        sum=1;ans=0;        for(i=1;i<=n;i++)        {            scanf("%lf%lf",&p[i].x,&p[i].y);        }        cnt=1;        for(i=1;i<n;i++)        {            for(j=i+1;j<=n;j++)            {                r[cnt].f=i;                r[cnt].t=j;                r[cnt].w=sqrt(fabs(pow(p[i].x-p[j].x,2)+pow(p[i].y-p[j].y,2)));                cnt++;                            }        }        //cout<<cnt<<endl;        sort(r+1,r+cnt,cmp);        Kruskal();    }    return 0;}