10397 - Connect the Campus

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#define MAX 600000
using namespace std;
struct node
{
    long long u,v;
    double cost;
} arr[MAX];
long long par[MAX];
long long khoj_rep(long long r)
{
    if(par[r]==r)
    {
        return r;
    }
    else
    {
        return par[r]=khoj_rep(par[r]);
    }
}
bool cmp(node lhs,node rhs)
{
    return lhs.cost<rhs.cost;
}
double arr1[MAX],arr2[MAX];
int main()
{
    long long koyta_building,koyta_cable,i,j,k,l,m,a,b,u,v;
    double x,y,z,sum;
    while(cin>>koyta_building)
    {
        for(i=1; i<=koyta_building; i++)
        {
            par[i]=i;
            cin>>arr1[i]>>arr2[i];
        }
        cin>>koyta_cable;
        for(l=0; l<koyta_cable; l++)
        {
            cin>>a>>b;
            u=khoj_rep(a);
            v=khoj_rep(b);
            if(u!=v)
            {
                par[u]=v;
            }
        }
        m=0;
        for(j=1; j<=koyta_building; j++)
        {
            for(k=j+1; k<=koyta_building; k++)
            {
                x=pow((arr1[j]-arr1[k]),2);
                y=pow((arr2[j]-arr2[k]),2);
                z=sqrt(x+y);
                arr[m].u=j;
                arr[m].v=k;
                arr[m].cost=z;
                m++;
            }
        }
        sort(arr,arr+m,cmp);
        sum=0.0;
        for(l=0; l<m; l++)
        {
            u=khoj_rep(arr[l].u);
            v=khoj_rep(arr[l].v);
            if(u!=v)
            {
                par[u]=v;
                sum+=arr[l].cost;
            }
        }
        printf("%.2lf\n",sum);
        memset(par,0,sizeof(par));
    }
    return 0;
}