UVA 11228 - Transportation system.

Click here to view the problem
/***
Md. Namzul Hasan
Shahjalal University of Science & Technology,Sylhet.
hasan08sust@gmail.com
***/
#include<bits/stdc++.h>
using namespace std ;
typedef long long ll ;
typedef int in ;
typedef unsigned long long ull ;
const double pi = 2*acos(0) ;
#define maxi 40000
#define pf printf
#define sc scanf
#define pb push_back
#define MEM(x,y) (memset((x),(y),sizeof(x)))
#define MIN(x,y) ((x) < (y) ? (x) : (y))
#define MAX(x,y) ((x) > (y) ? (x) : (y))
#define load(array,size)  for(int i=0 ; i<size ; i++) cin>>array[i]  ;
#define new_line  pf("\n")
#define clear_data(array) memset(array,0,sizeof(array))
#define highest_int 2147483647

struct edge
{
    int u,v ;
    double w ;
};

double x[10001] , y[10001] ;
vector<edge> e ;
double r1, r2 ;
int c1, c2 ;
int pr[10001] ;

double dis(int i, int j)
{
    double d = sqrt( (x[i]-x[j]) * (x[i]-x[j]) + (y[i]-y[j]) * (y[i]-y[j]) ) ;
    return d ;
}

int find_r(int r)
{
        return pr[r]==r ? r : find_r(pr[r]) ;
}

bool comp(edge a, edge b)
{
    return a.w< b.w ;
}

void mst(int n, int r)
{
    sort(e.begin(),e.end(),comp) ;
    r1 =0 ;
    r2 =0 ;
    c1 =0 ;
    c2 =0 ;
    int i,u,v,a1,a2 ;
    for(i=0 ; i <n ; i++)
        pr[i] =i ;

    for(i=0 ; i<e.size() ; i++)
    {
        u = find_r(e[i].u) ;
        v = find_r(e[i].v) ;
        if(u!=v)
        {
            if(e[i].w<=r)
            {
                r1+=e[i].w ;
                c1++ ;
            }
            else
            {
                r2+=e[i].w ;
                c2++ ;
            }
            pr[u] = v ;
        }
    }

    a1 = round(r1) ;
    a2 = round(r2) ;
    pf("%d %d %d\n",c2+1,a1,a2) ;

}

int main()
{
    int test ;
    sc("%d",&test) ;
    int n ,r,i,j ;
    int cnt =1 ;

    while(test--)
    {
        e.clear() ;

        sc("%d %d",&n,&r) ;
        for(i=0 ; i<n ; i++)
           cin>>x[i]>>y[i] ;

        edge get ;
        for(i=0 ; i<n ; i++)
        {
            for(j =i+1 ; j<n ; j++)
            {
                get.u =i ;
                get.v = j ;
                get.w = dis(i,j) ;
                e.pb(get) ;
            }
        }

        pf("Case #%d: ",cnt++) ;
        mst(n,r) ;
    }
    return 0;
}