10分kruskal求助
查看原帖
10分kruskal求助
477737
quyy06楼主2021/8/22 09:18
#include<cmath>
#include<cstdio>
#include<algorithm>
using namespace std;
int n,m,cnt,cnt1,fa[5000100];double ans;
struct point{
    int x,y;
}po[5000100];
struct node{
    int u,v;
    double w;
}a[5500005];
int find(int now){return fa[now]==now?now:fa[now]=find(fa[now]);}
double line(int l,int r){return (double)(sqrt((double)(po[l].x-po[r].x)*(po[l].x-po[r].x)+(double)(po[l].y-po[r].y)*(po[l].y-po[r].y)));}
bool cmp(node l,node r){return l.w<r.w;}
void kruskal(){
    for(int i=1;i<=cnt;i++){
        int un=find(a[i].u),vn=find(a[i].v);
        if(un==vn) continue;
        fa[vn]=un;
        ans+=a[i].w;
        if(++cnt1==n-1) return;
    }
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) fa[i]=i;
    for(int i=1;i<=n;i++) scanf("%d%d",&po[i].x,&po[i].y);
    for(int i=1;i<=n;i++)
        for(int j=i+1;j<=n;j++){
            a[++cnt].u=i;
            a[cnt].v=j;
            a[cnt].w=line(i,j);
        }
    for(int i=1;i<=m;i++){
        scanf("%d%d",&a[++cnt].u,&a[cnt].v);
        a[cnt].w=0.0;
    }
    sort(a+1,a+1+cnt,cmp);
    kruskal();
    printf("%.2lf\n",ans);
    return 0;
}
2021/8/22 09:18
加载中...