Cod sursa(job #2508989)

Utilizator AlexAboAbogatoaie Alexandru AlexAbo Data 13 decembrie 2019 16:29:34
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 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,x,c,y,rx,ry,t[N],cost;
priority_queue<tuple<int,int,int>> edges;
vector<pair<int,int>>sol,v[N];
void add(int);
bitset<N> viz;
int main()
{
    f>>n>>m;
    for(;m;m--)
    {
        f>>x>>y>>c;
        v[x].push_back(make_pair(c,y));
        v[y].push_back(make_pair(c,x));
    }
    add(1);
    for(int cnt=n-1;cnt;cnt--)
    {
        do
        {
            tie(c,x,y)=edges.top();
            edges.pop();
        }
        while(viz[x]&&viz[y]);
        sol.push_back(make_pair(x,y));
        cost-=c;
        x=viz[x]?y:x;
        add(x);
    }
    g<<cost<<'\n'<<sol.size()<<'\n';
    for(auto it:sol)
        g<<it.first<<' '<<it.second<<'\n';
    return 0;
}
void add(int nod)
{
    viz[nod]=1;
    for(auto it:v[nod])
    {
        tie(c,x)=it;
        if(!viz[x])
            edges.push(make_tuple(-c,nod,x));
    }
}