11710 - Expensive subway

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#define MAX 80009
#include<map>
using namespace std;
map<string,int>mymap;
long long par[410];
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,k,cost,incr,sum,u,v,counter;
    string s,s1,s2,s3;
    while(cin>>M>>N)
    {
        if(M==0 && N==0)
        {
            break;
        }
        mymap.clear();
        for(i=1;i<=M;i++)
        {
            par[i]=i;
        }
        incr=0;
        for(i=0;i<M;i++)
        {
            cin>>s;
            if(!mymap[s])
            {
                mymap[s]=++incr;
            }
        }
        for(i=0;i<N;i++)
        {
            cin>>s1>>s2>>cost;
            arr[i].u=mymap[s1];
            arr[i].v=mymap[s2];
            arr[i].cost=cost;
        }
        cin>>s3;
        sort(arr,arr+N,cmp);
        sum=0;
        counter=0;
        for(j=0;j<i;j++)
        {
            u=khoj_rep(arr[j].u);
            v=khoj_rep(arr[j].v);
            if(u!=v)
            {
                par[u]=v;
                counter++;
                sum+=arr[j].cost;
            }
        }
        if(counter==M-1)
        {
            cout<<sum<<endl;
        }
        else
        {
            cout<<"Impossible"<<endl;
        }
        memset(par,0,sizeof(par));
    }
    return 0;
}

0 comments: (+add yours?)