UVA 1216 - The Bug Sensor Problem

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
#define maxi 100005

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

vector<edge> e ;
int pr[maxi] ;
vector<int> xx, yy ;
edge get ;

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

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

void mst(int n, int m)
{
    int i,cnt =n,u,v ;
    double sum = 0.0 ;

    for(i=0 ; i<n ; i++)
        pr[i] =i ;
    sort(e.begin(),e.end(),comp) ;

    bool flag =true ;
    int ln = e.size() ;

    for(i=0 ; i<e.size(); i++)
    {
        u =find_r(e[i].u) ;
        v = find_r(e[i].v) ;
        if(u!=v)
        {
            pr[u] =v ;
            sum = max(sum,e[i].w) ;
            cnt-- ;
        }
        if(cnt==m)
        {
            sum = e[i].w ;
            flag =false ;
            break ;
        }
    }

    if(flag)
    {
        sum = e[ln-1].w ;
    }

    pf("%d\n",(int)ceil(sum)) ;

}


int main()
{
    int test,m,p,n,i,j ;
    sc("%d",&test) ;

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

        sc("%d",&m) ;
        while(1)
        {
            sc("%d",&p) ;
            if(p==-1)
                break ;
            xx.pb(p) ;
            sc("%d",&p) ;
            yy.pb(p) ;
        }
        n = xx.size() ;
        double dis ;

        for(i=0 ; i<n ; i++)
        {
            for(j=i+1 ; j<n ; j++)
            {
                dis =hypot((xx[i]-xx[j]),(yy[i]-yy[j])) ;
                get.u =i ;
                get.v =j ;
                get.w =dis ;
                e.pb(get) ;
            }
        }

        mst(n,m) ;
    }

    return 0;
}