Cod sursa(job #3196547)

Utilizator Emilia23Dobra Emilia Emilia23 Data 24 ianuarie 2024 11:02:08
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.07 kb
#include <bits/stdc++.h>

using namespace std;

ifstream f("apm.in");
ofstream g("apm.out");

int n,m,s,k,tata[200005],dim[200005];
struct elem
{
    int x,y,c;
}v[400005],sol[400005];

inline bool cmp(const elem &a,const elem &b)
{
    return a.c<b.c;
}

int tata_multime(int x)
{
    if(x!=tata[x])
        return tata_multime(tata[x]);
    return tata[x];
}

void unire(int x,int y)
{
    x=tata_multime(x);
    y=tata_multime(y);
    if(dim[x]<dim[y])
    {
        dim[y]+=dim[x];
        tata[x]=y;
    }
    else
    {
        dim[x]+=dim[y];
        tata[y]=x;
    }
}

int main()
{
    int i,j;
    f>>n>>m;
    for(i=1;i<=m;i++)
        f>>v[i].x>>v[i].y>>v[i].c;
    sort(v+1,v+m+1,cmp);
    for(i=1;i<=n;i++)
        tata[i]=i,dim[i]=1;
    for(i=1;i<=m && k<n-1;i++)
    {
        if(tata_multime(v[i].x)!=tata_multime(v[i].y))
        {
            unire(v[i].x,v[i].y);
            s+=v[i].c;
            sol[++k]=v[i];
        }
    }
    g<<s<<'\n'<<n-1<<'\n';
    for(i=1;i<=k;i++)g<<sol[i].x<<' '<<sol[i].y<<'\n';
}