10147 - Highways

#include<cstdio>
#include<cmath>
#include<cstring>
#include<iostream>
#include<algorithm>
#define MAX 600000
using namespace std;
long long par[800];
long long arr1[800];
long long arr2[800];
struct node
{
    long long x,y;
    double cost;
}arr[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;
}
int main()
{
    long long test,koyta_highway,built_in,i,j,k,x,y,u,v,yes,tag=0;
    double a,b,c;
    cin>>test;
    while(test--)
    {
        cin>>koyta_highway;
        if(tag)
        {
            cout<<endl;
        }
        tag=1;
        for(i=1;i<=koyta_highway;i++)
        {
            par[i]=i;
        }
        for(i=1;i<=koyta_highway;i++)
        {
            cin>>arr1[i]>>arr2[i];
        }
        cin>>built_in;
        for(i=1;i<=built_in;i++)
        {
            cin>>x>>y;
            u=khoj_rep(x);
            v=khoj_rep(y);
            if(u!=v)
            {
                par[u]=v;
            }
        }
        k=0;
        for(i=1;i<=koyta_highway;i++)
        {
            for(j=i+1;j<=koyta_highway;j++)
            {
                a=pow((arr1[i]-arr1[j]),2);
                b=pow((arr2[i]-arr2[j]),2);
                c=sqrt(a+b);
                arr[k].x=i;
                arr[k].y=j;
                arr[k].cost=c;
                k++;
            }
        }
        sort(arr,arr+k,cmp);
        yes=0;
        for(i=0;i<k;i++)
        {
            u=khoj_rep(arr[i].x);
            v=khoj_rep(arr[i].y);
            if(u!=v)
            {
                yes=1;
                par[u]=v;
               cout<<arr[i].x<<" "<<arr[i].y<<endl;
            }
        }
        if(!yes)
        {
            cout<<"No new highways need"<<endl;
        }
        memset(par,0,sizeof(par));
    }
    return 0;
}

0 comments: (+add yours?)