Cod sursa(job #1857305)

Utilizator SkiryFarauanu Ionut Skiry Data 26 ianuarie 2017 00:25:27
Problema Arbore partial de cost minim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.98 kb
#include <bits/stdc++.h>

using namespace std;

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

const int N=200010;
int n, m, i, j, t[N], x, y, c, cost, a, b, compr(int );
pair<int, pair<int,int> > M[2*N];
struct nod
{
    int nr,val;
}p;
vector<nod> sol;
int main()
{
    f>>n>>m;

    for(i=1; i<=n; i++)
        t[i]=i;
    for(i=1;i<=m;i++)
    {
        f>>x>>y>>c;
        M[i]={c,{x,y}};
    }
    sort(M+1, M+m+1);
    for(i=1,j=n-1; j; i++)
    {
        a=M[i].second.first;
        b=M[i].second.second;
        x=compr(a);
        y=compr(b);
        if(x!=y)
        {
            t[x]=y;
            cost+=M[i].first;
            p.nr=a;
            p.val=b;
            sol.push_back(p);
            j--;
        }
    }
     g<<cost<<'\n'<<n-1<<'\n';
    for(i=0;i<sol.size();i++)
        g<<sol[i].val<<" "<<sol[i].nr<<'\n';
    return 0;
}
int compr(int v)
{
    if(t[v]==v)
        return v;
    t[v]=compr(t[v]);
    return t[v];
}