1. 程式人生 > >暢通工程再續-最小生成樹+並查集

暢通工程再續-最小生成樹+並查集

\n pla 數據 float turn 全部 一位 clu back

原題鏈接:https://vjudge.net/problem/15740/origin

相信大家都聽說一個“百島湖”的地方吧,百島湖的居民生活在不同的小島中,當他們想去其他的小島時都要通過劃小船來實現。現在政府決定大力發展百島湖,發展首先要解決的問題當然是交通問題,政府決定實現百島湖的全暢通!經過考察小組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!

題解:這題和之前的稍有不同,要先把點轉化為邊,這個題要用double來算,我之前被坑了一波.最小花費用最小生成樹,能否暢通用並查集.

#include <bits/stdc++.h>
#define inf 0x3f3f3f
#define float double
const int N=105;
using namespace std;
struct pos{
    int x,y;
    //pos(int _x,int _y):x(_x),y(_y){}
};
struct road{
    int v;
    float w;
    road(int _v,float _w):v(_v),w(_w){}
};
pos p[N];
int se[N];
int mp[N];
int n;
vector<road> G[N];
int fd(int a){
    int x=a;
    while(se[x]!=x){
        x=se[x];
        for(int i=a,j;i!=x;i=j){
            j=se[i];
            se[i]=x;
        }
    }
    return x;
}
void join(int a,int b){
    int fa=fd(a);
    int fb=fd(b);
    se[fa]=fb;
}
float bfs(){
    memset(mp,0,sizeof(mp));
    vector<int> vr;
    vr.push_back(1);
    mp[1]=1;
    //unsigned nn=n;
    float ans=0;
    while(vr.size()!=n){
        float minn=inf;
        int id;
        for(unsigned i=0;i<vr.size();i++){
            for(auto x:G[vr[i]]){
                if(mp[x.v]==0 && minn>x.w) minn=x.w,id=x.v;
            }
        }
        vr.push_back(id);
        mp[id]=1;
        ans+=minn;
    }
    return ans;
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--){
        memset(G,0,sizeof(G));
        scanf("%d",&n);
        for(int i=1;i<=n;i++){
            scanf("%d%d",&p[i].x,&p[i].y);
            se[i]=i;
        }
        //cout<<"kkkk"<<endl;
        //for(int i=1;i<=n;i++) se[i]=i;
        for(int i=1;i<=n;i++){
            for(int j=i+1;j<=n;j++){
                //if(i==j) continue;
                float sqrtn=sqrt(abs(p[i].x-p[j].x)*abs(p[i].x-p[j].x)+abs(p[i].y-p[j].y)*abs(p[i].y-p[j].y));
                if(sqrtn>=10 && sqrtn<=1000){
                    join(i,j);
                    G[i].emplace_back(j,sqrtn);
                    G[j].emplace_back(i,sqrtn);
                }
            }
        }
       // cout<<"lll"<<endl;
        int ans=0;
        for(int i=1;i<=n;i++){
            if(se[i]==i) ans++;
        }
        if(ans>1){
            printf("oh!\n");
            continue;
        }
        printf("%.1f\n",100*bfs());

    }
    //cout << "Hello world!" << endl;
    return 0;
}

暢通工程再續-最小生成樹+並查集