Hdu1875 畅通工程再续

畅通工程再续 Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others) Total Submission(s): 41638    Accepted Submission(s): 13980   Problem 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!

Kruskal：

#include<iostream>
#include<cmath>
#include<algorithm>
#include<iomanip>
#define N 101
using namespace std;
int pre[N],t,n;
struct di
{
    int l,r;
}a[N];
struct node
{
    double x,y;
    double w;
}edge[10001];
int cmp(node a,node b)
{
    return a.w<b.w;
}
int find(int x)
{
    int p=x,temp;
    while(x!=pre[x])
        x=pre[x];
    while(p!=x)
    {
        temp=pre[p];
        pre[p]=x;
        p=temp;
    }
    return x;
}
void merge(int a,int b)
{
    a=find(a);
    b=find(b);
    if(a!=b)
        pre[b]=a;
}
double kruskal(int k)
{
        double ans=0;
        sort(edge,edge+k,cmp);
        for(int i=0;i<k;i++)
        {
            int a=find(edge[i].x);
            int b=find(edge[i].y);
            if(a!=b)
            {
                if(edge[i].w>=10&&edge[i].w<=1000)
                {
                ans+=edge[i].w;
                merge(a,b);
                }
            }
        }
        return ans;
}
int main()
{
    while(cin>>t&&t>0&&t<=200)
    {
        while(t--)
        {
            cin>>n;
            if(n<0||n>200)
                break;
            for(int i=1;i<=n;i++)
                pre[i]=i;
            for(int i=1;i<=n;i++)
            {
                cin>>a[i].l>>a[i].r;
            }
            int k=0;
            for(int i=1;i<=n;i++)
                for(int j=i+1;j<=n;j++)
            {
                edge[k].x=i;
                edge[k].y=j;
                edge[k].w=sqrt((a[i].l-a[j].l)*(a[i].l-a[j].l)+(a[i].r-a[j].r)*(a[i].r-a[j].r));
                k++;
            }
            double sum=kruskal(k);
            int cnt=0;
            for(int i=1;i<=n;i++)
            {
                if(pre[i]==i)
                    cnt++;
            }
            if(cnt==1)
                cout<<fixed<<setprecision(1)<<sum*100<<endl;
            else
                cout<<"oh!"<<endl;
        }
    }
    return 0;
}

Prim：

#include<iostream>
#include<cstring>
#include<iomanip>
#include<cmath>
#define inf 0x3f3f3f3f
#define N 101
using namespace std;
int vis[N];
int t,n;
double map[N][N],dis[N];
struct di
{
    int x,y;
}a[N];
double prim(int n)
{
        for(int i=1;i<=n;i++)
            dis[i]=inf;
        double ans=0;
        for(int j=0;j<=(n*(n-1))/2;j++)
        {
            double t=inf;
            int pos;
            dis[1]=0;
            for(int i=1;i<=n;i++)
            {
                if(!vis[i]&&t>dis[i])
                {
                    t=dis[i];
                    pos=i;
                }
            }
            vis[pos]=1;
            for(int i=1;i<=n;i++)
            {
                if(!vis[i]&&map[pos][i]!=inf&&dis[i]>map[pos][i]&&map[pos][i]<=1000.000&&map[pos][i]>=10.000)
                        dis[i]=map[pos][i];
            }
        }
        for(int i=1;i<=n;i++)
        {
            if(dis[i]>=10.000&&dis[i]<=1000.000)
                ans+=dis[i];
        }
        return ans;
}
int main()
{
    while(cin>>t&&t<=200&&t>0)
    {
        while(t--)
        {
            cin>>n;
            if(n<0||n>100)
                break;
            memset(vis,0,sizeof(vis));
            memset(map,inf,sizeof(map));
            for(int i=1;i<=n;i++)
            {
                cin>>a[i].x>>a[i].y;
            }
            int k=0;
            for(int i=1;i<=n;i++)
            {
                for(int j=1;j<=n;j++)
                {
                    map[i][j]=sqrt((a[i].x-a[j].x)*(a[i].x-a[j].x)+(a[i].y-a[j].y)*(a[i].y-a[j].y));
                }
            }
            double sum=prim(n);
            int flag=1;
            for(int i=1;i<=n;i++)
            {
                if(vis[i]==0)
                    flag=0;
            }
            if(flag)
                cout<<fixed<<setprecision(1)<<sum*100<<endl;
            else
                cout<<"oh!"<<endl;
        }
    }
    return 0;
}