Cod sursa(job #2200613)

Utilizator PushkinPetolea Cosmin Pushkin Data 1 mai 2018 23:13:01
Problema Arbore partial de cost minim Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.29 kb
#include <iostream>
#include <bits/stdc++.h>
#include <fstream>
#include <climits>
using namespace std;
ifstream f("kruskal.in");
ofstream g("kruskal.out");
struct muchie
{
    int x, y, c;
    bool operator<(muchie m)
    {
        return(c<m.c);
    }
};
vector<muchie> G;
vector<int> t;
int nc, n, m, j, c, cst;
void rd()
{
    f>>n>>m;
    G.resize(m);
    t.resize(n+1);
    for(int i=0;i<m;i++)
        f>>G[i].x>>G[i].y>>G[i].c;
}
inline int tata(int x)
{
    while(t[x])
        x=t[x];
    return x;
}
inline void unite(int x, int y)
{
    t[tata(x)]=tata(y);
}
inline void compresie(int x, int r)
{
    int cx;
    while(t[x])
    {
        cx=t[x];
        t[x]=r;
        x=cx;
    }
}
inline bool q(int x, int y)
{
    int r=tata(x);
    compresie(x, r);
    return(r==tata(y));
}
void Kruskal()
{
    vector<muchie>::iterator i=G.begin();
    nc=n;
    sort(G.begin(), G.end());
    while(nc>1)
    {
        while(q(i->x, i->y))
            G.erase(i);
        unite(i->x, i->y);
        cst+=i->c;
        i++;
        nc--;
    }
    G.erase(i, G.end());
}
void afis()
{
    g<<cst<<'\n'<<G.size()<<'\n';
    for(auto a:G)
        g<<a.x<<' '<<a.y<<'\n';
}
int main()
{
    rd();
    Kruskal();
    afis();
    return 0;
}