Cod sursa(job #2045470)

Utilizator silviuilasSilviu Ilas silviuilas Data 22 octombrie 2017 13:50:59
Problema Arbore partial de cost minim Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 0.94 kb
#include <bits/stdc++.h>
using namespace std;
ifstream f("apm.in");
ofstream g("apm.out");
const int N = 200010;
vector<int> v[N];
vector<pair<int,int>> sol;
vector<tuple<int,int,int>> e;
int n,m,x,y,c,cost,t[N],rx,ry,i,get_root(int);
queue<int> q;
int main()
{
    f>>n>>m;
    for(i=1;i<=m;i++)
    {
        f>>x>>y>>c;
        e.push_back(make_tuple(c,x,y));
    }
    sort(e.begin(),e.end());
    for(i=1;i<=n;i++)
        t[i]=i;
    for(auto it:e)
    {
        tie(c,x,y)=it;
        rx=get_root(x);
        ry=get_root(y);
        if(rx==ry)
            continue;
        sol.push_back(make_pair(x,y));
        cost+=c;
        if(int(sol.size())==n-1)
            break;
    }
    g<<cost<<'\n'<<n-1<<'\n';
    for(auto it:sol)
        g<<it.first<<' '<<it.second<<'\n';
    return 0;
}
int get_root(int nod)
{
    if(t[nod]==nod)
        return nod;
    t[nod]=get_root(t[nod]);
    return t[nod];
}