Cod sursa(job #2652515)

Utilizator teisanumihai84Mihai Teisanu teisanumihai84 Data 25 septembrie 2020 06:11:25
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.17 kb
#include <fstream>
#include <algorithm>
#define DIM 200001
using namespace std;
ifstream fin ("apm.in");
ofstream fout ("apm.out");
struct muchie {
    int x;
    int y;
    int c;
};
muchie v[2*DIM], sol[DIM];
int n, m, t[DIM], i, S, dim;
bool cmp (muchie i, muchie j)
{
    return i.c<j.c;
}
int rad (int nod)
{
    while (t[nod]>0)
        nod=t[nod];
    return nod;
}
int main()
{
    fin>>n>>m;
    for (i=1; i<=m; i++)
        fin>>v[i].x>>v[i].y>>v[i].c;
    sort (v+1, v+m+1, cmp);
    for (i=1; i<=n; i++)
        t[i]=-1;
    // retinem cu -
    for (i=1; i<=m; i++)
    {
        int rx=rad(v[i].x);
        int ry=rad(v[i].y);
        if (rx!=ry)
        {
            if (-t[rx]>-t[ry])
            {
                t[rx]+=t[ry];
                t[ry]=rx;
            }
            else
            {
                t[ry]+=t[rx];
                t[rx]=ry;
            }
            S+=v[i].c;
            sol[++dim].x=v[i].x;
            sol[dim].y=v[i].y;
        }
        if (dim==n-1)
            break;
    }
    fout<<S<<"\n"<<n-1<<"\n";
    for (i=1; i<=dim; i++)
        fout<<sol[i].x<<" "<<sol[i].y<<"\n";
}