Cod sursa(job #3189665)

Utilizator Alexinfo22Rusu Luca Alexinfo22 Data 6 ianuarie 2024 13:23:07
Problema Arbore partial de cost minim Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.18 kb
#include <bits/stdc++.h>

using namespace std;
ifstream fin("apm.in");
ofstream fout("apm.out");
int n,m,tata[200001],cnt,Rang[200001],s;
struct pereche
{
    int a,b,cost;
    bool ok;
} x[400001];
bool cmp(pereche a,pereche b)
{
    return a.cost<b.cost;
}
int Radacina(int k)
{
    while(tata[k]!=k)
        k=tata[k];
    return k;
}
void Compresie(int p,int q)
{
    int r=p,s=q;
    if(r!=s)
    {
        if(Rang[r]>Rang[s])
            tata[s]=r;
        else
        {
            tata[r]=s;
            if(Rang[r]==Rang[s])
                Rang[r]++;
        }
    }
}
vector<pair<int,int>>v[100001];
int main()
{
    fin>>n>>m;
    for(int i=1; i<=m; i++)
        fin>>x[i].a>>x[i].b>>x[i].cost;
    sort(x+1,x+m+1,cmp);
    for(int i=1; i<=n; i++)
        tata[i]=i;
    for(int i=1; i<=m; i++)
    {
        if(Radacina(x[i].a)!=Radacina(x[i].b))
        {
            s+=x[i].cost;
            Compresie(Radacina(x[i].a),Radacina(x[i].b));
            cnt++;
            x[i].ok=1;
        }
    }
    fout<<s<<'\n'<<cnt<<'\n';
    for(int i=1; i<=m; i++)
        if(x[i].ok)
        fout<<x[i].a<<' '<<x[i].b<<'\n';
    return 0;
}