Cod sursa(job #1340332)

Utilizator Pintilie_AndreiFII-Pintilie Andrei Pintilie_Andrei Data 11 februarie 2015 19:21:50
Problema Arbore partial de cost minim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.46 kb
#include <iostream>
#include <fstream>
#include <algorithm>
#include <vector>
#define N 200005
#define oo 2000000000
using namespace std;
int n,m;
struct str
{
    int x,y,c;
};
int sol[2][N];
int sors[N];
int compcon[N];
int suma,top;
vector <str> a;

bool Comp(str g,str h)
{
    return g.c<h.c;
}

void Read()
{
    ifstream fin("apm.in");
    fin>>n>>m;
    int xx,yy,cost,i;
    str w;
    for( i=1; i<=m; i++)
    {
        fin>>w.x>>w.y>>w.c;
        a.push_back(w);
    }
    sort(a.begin(),a.begin()+a.size(),Comp);

}

int GetGroup(int nr)
{
    if(nr==compcon[nr])
        return nr;
    compcon[nr]=GetGroup(compcon[nr]);
    return compcon[nr] ;
}
void Unite(int good, int bad)
{
    compcon[GetGroup(bad)]=GetGroup(good);
}

void Adauga(int i)
{
    top++;
    sol[0][top]=a[i].x;
    sol[1][top]=a[i].y;
    suma+=a[i].c;
}

void Ini()
{
    for(int i=1; i<=n; i++)
        compcon[i]=i;
}
void Salv()
{
    Ini();
   int nrm=n-1,compx,compy;
   int i=0;
   int nr=0;
   while(nrm && i<m)
   {
       compx=GetGroup(a[i].x);
       compy=GetGroup(a[i].y);
        if(compx!=compy)
        {
            Unite(a[i].x,a[i].y);
            Adauga(i);
            nrm--;
        }
       i++;
   }
}
int main()
{
    ofstream fout("apm.out");
   Read();
   Salv();
   fout<<suma<<"\n"<<top<<"\n";
   for(int i=1; i<=top; i++)
    fout<<sol[0][i]<<" "<<sol[1][i]<<"\n";

    return 0;
}