11733 - Airports

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define MAX 100009
using namespace std;
long long par[MAX];
struct node
{
    long long u,v,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 a,test,koyta_location,koyta_road,airport_cost,u,v,i,j,k,tag=0,airport,minimum,counter;
    cin>>test;
    while(test--)
    {
        cin>>koyta_location>>koyta_road>>airport_cost;
        for(j=1;j<=koyta_location;j++)
        {
            par[j]=j;
        }
        for(i=0; i<koyta_road; i++)
        {
            cin>>arr[i].u>>arr[i].v>>arr[i].cost;
        }
        sort(arr,arr+i,cmp);
        minimum=0;
        airport=0;
        counter=0;
        for(k=0; k<i; k++)
        {
            u=khoj_rep(arr[k].u);
            v=khoj_rep(arr[k].v);
            if(u!=v && arr[k].cost<airport_cost)
            {
                par[u]=v;
                minimum+=arr[k].cost;
            }
            else if(u!=v && arr[k].cost>=airport_cost)
            {
                par[u]=v;
                counter++;
            }
        }
        for(k=1;k<=koyta_location;k++)
        {
            if(par[k]==k)
            {
                airport++;
            }
        }
        airport+=counter;
        a=airport*airport_cost;
        minimum+=a;
        printf("Case #%lld: %lld %lld\n",++tag,minimum,airport);
        memset(par,0,sizeof(par));
    }
    return 0;
}

0 comments: (+add yours?)