Cod sursa(job #2369775)

Utilizator andrei32576Andrei Florea andrei32576 Data 6 martie 2019 09:11:44
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.94 kb
#include <bits/stdc++.h>
using namespace std;

int n,m,i,x[200002],y[200002],c[200002],ct,k,v[200002];
int gr[200002],ind[200002];

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

bool cmp(int i,int j)
{
    return c[i]<c[j];
}

int grupa(int i)
{
    if(gr[i]==i) return i;
    gr[i]=grupa(gr[i]);
    return gr[i];
}

void reuniune(int i,int j)
{
    gr[grupa(i)]=grupa(j);
}

int main()
{
    f>>n>>m;

    for(i=1;i<=m;i++)
    {
        f>>x[i]>>y[i]>>c[i];
        ind[i]=i;
    }

    sort(ind+1,ind+m+1,cmp);

    for(i=1;i<=n;i++)
        gr[i]=i;

    ct=0;
    for(i=1;i<=m;i++)
    {
        if(grupa(x[ind[i]])!=grupa(y[ind[i]]))
        {
            ct+=c[ind[i]];
            reuniune(x[ind[i]],y[ind[i]]);
            v[++k]=ind[i];
        }
    }

    g<<ct<<"\n"<<n-1<<"\n";
    for(i=1;i<=k;i++)
        g<<x[v[i]]<<" "<<y[v[i]]<<"\n";

    f.close();
    g.close();
    return 0;
}