Cod sursa(job #2532775)

Utilizator Bogdan2728Iamnitchi Bogdan Bogdan2728 Data 28 ianuarie 2020 12:33:10
Problema Arbore partial de cost minim Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.3 kb
#include <bits/stdc++.h>
#define Nmax 200000
#define Mmax 400000
using namespace std;

int n,m,total,k,tata[Nmax],rang[Nmax];

struct Muchie{
    int x,y,c;
}v[Mmax];

pair<int,int> p[Nmax];

void Read()
{
    ifstream f("apm.in");
    f>>n>>m;
    for(int i=1;i<=m;i++)
    {
        f>>v[i].x>>v[i].y>>v[i].c;
    }

    for(int i=1;i<=n;i++)
    {
        tata[i]=i;
        rang[i]=1;
    }
    f.close();
}

int Find(int nod)
{
    while(nod != tata[nod])
        nod = tata[nod];
    return nod;
}

void Unire(int x,int y)
{
    if(rang[x] < rang[y])
        tata[x] = y;
    if(rang[x] > rang[y])
        tata[y] = x;
    if(rang[x]==rang[y])
    {
        tata[x]=y;
        rang[x]++;
    }
}

void Afisare()
{
    ofstream g("apm.out");
    g<<total<<"\n";
    g<<k<<"\n";
    for(int i=1;i<=k;i++)
        g<<p[i].first<<" "<<p[i].second<<"\n";
    g.close();
}

void Solve()
{
    for(int i=1;i<=m;i++)
    {
        int tata_x = Find(v[i].x);
        int tata_y = Find(v[i].y);
        if(tata_x != tata_y)
        {
            Unire(tata_x,tata_y);
            p[++k].first = v[i].x;
            p[k].second = v[i].y;
            total += v[i].c;
        }
    }
}
int main()
{
    Read();
    Solve();
    Afisare();
    return 0;
}