HDU 1875 暢通工程再續 Kruskal求解

題目鏈接:http://acm.hust.edu.cn/vjudge/contest/view.action?cid=66965#problem/N

題意:給點每個點的座標及連通條件,求最小代價。

這題真的傻逼了,我居然,直接把距離平方進行相加,真是醉了。。。。其他沒啥值的注意的地方。

PS:看到有人double的輸出用的是%lf也可以AC,但是,還是的注意一下,double的輸出使用%f,有的題WA過。


代碼:

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>

using namespace std;
struct edge{
    int u,v;
    double w;
    bool operator<(const edge& a)const{
        return w<a.w;
    }
}E[10010];

int X[110],Y[110];
int C,tot;
int pre[110];

int find(int x){
    int t=x;
    while(t!=pre[t]) t=pre[t];
    while(x!=t) pre[x]=t,x=pre[x];
    return t;
}

void Kruskal(){
    for(int i=0;i<=C;i++) pre[i]=i;
    int cnt=1;
    double sum=0;
    for(int i=0;i<tot;i++){
        int u=E[i].u,v=E[i].v;
        int fu=find(u),fv=find(v);
        if(fu==fv) continue;
        pre[fv]=fu;sum+=E[i].w;cnt++;
        if(cnt==C) break;
    }
    if(cnt<C) cout<<"oh!\n";
    else printf("%.1f\n",sum*100);
}

int main(){
    //freopen("D:\\in.txt","r",stdin);
    int T;cin>>T;
    while(T--){
        cin>>C;
        for(int i=0;i<C;i++)
            scanf("%d %d",X+i,Y+i);
        tot=0;
        for(int i=0;i<C;i++) for(int j=i+1;j<C;j++){
            int t=(X[i]-X[j])*(X[i]-X[j])+(Y[i]-Y[j])*(Y[i]-Y[j]);
            if(t<100 || t>1000000) continue;
            E[tot++]=(edge){i,j,sqrt(t)};
        }
        sort(E,E+tot);
        Kruskal();
    }
    return 0;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章