Cod sursa(job #2081007)

Utilizator MihalachiRazvanMihalachi Razvan MihalachiRazvan Data 3 decembrie 2017 19:36:08
Problema Arbore partial de cost minim Scor 60
Compilator cpp Status done
Runda Arhiva educationala Marime 1.46 kb
#include<vector>
#include<algorithm>
#include <fstream>
#include <iostream>
#define maxim 200000
using namespace std;
ifstream fin("apm.in");
ofstream fout("apm.out");
int n,m,s=0,x,c,y,a[maxim],k=0;
std::vector <pair<int,pair<int,int> > > vact;
std::vector <int> vect1[maxim];
std::vector <pair<int,int> >vect;
int main()
{
    fin>>n>>m;
    for(int i=1;i<=n;i++)
        a[i]=i;
    for(int i=1;i<=m;i++)
    {
        fin>>x>>y>>c;
        vact.push_back(make_pair(c,make_pair(x,y)));
    }
             for(int i=1;i<=n;i++)
        a[i]=i;
    sort(vact.begin(),vact.end());
    int s=0;
    for(int i=0;i<m;i++)
    {
      x=vact[i].second.first;
      y=vact[i].second.second;
      int l=vect1[a[x]].size();
      int aux=a[a[y]],aux1;
      if(a[x]!=a[y])
        {
            vect.push_back(make_pair(x,y));
            if(vect1[a[y]].size()!=0)
            {
                vect1[a[x]].push_back(aux);
                vect1[a[x]].insert(vect1[a[x]].end(),vect1[a[y]].begin(),vect1[a[y]].end());
                for(int j=0;j<vect1[a[x]].size();j++)
                    a[vect1[a[x]][j]]=a[x];
                vect1[aux].clear();
            }
            else
            {vect1[a[x]].push_back(y);
             a[y]=a[x];}
            s=s+vact[i].first;

    }
    }
    fout<<s<<"\n";
    fout<<vect.size()<<"\n";
    for(int i=0;i<vect.size();i++)
     fout<<vect[i].first<<"  "<<vect[i].second<<"\n";
    return 0;
}