11631 - Dark roads

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#define MAX 200009
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 m,n,i,j,sum,sum1,save_amount,u,v;
    while(scanf("%lld%lld",&m,&n)==2)
    {
        if(m==0&&n==0)
        {
            break;
        }
        for(i=1;i<=m;i++)
        {
            par[i]=i;
        }
        sum=0;
        for(j=0;j<n;j++)
        {
            scanf("%lld%lld%lld",&arr[j].u,&arr[j].v,&arr[j].cost);
            sum+=arr[j].cost;
        }
        sort(arr,arr+n,cmp);
        sum1=0;
        for(i=0;i<j;i++)
        {
            u=khoj_rep(arr[i].u);
            v=khoj_rep(arr[i].v);
            if(u!=v)
            {
                par[u]=v;
                sum1+=arr[i].cost;
            }
        }
        save_amount=sum-sum1;
        printf("%lld\n",save_amount);
        memset(par,0,sizeof(par));
    }
    return 0;
}

0 comments: (+add yours?)