Cod sursa(job #2698251)

Utilizator DesertChuStefan Andrei DesertChu Data 21 ianuarie 2021 16:08:45
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.16 kb
#include <bits/stdc++.h>
#define ft first
#define sd second
#define pb push_back
using namespace std;
ifstream f("apm.in");
ofstream g("apm.out");
priority_queue<pair<int,pair<int,int> >, vector<pair<int,pair<int,int> > >, greater<pair<int,pair<int,int> > > >h;// in ordine: cost, nod1, nod2
vector<pair<int,int> >v[200003];
vector<pair<int,int> >sol;
int n,m,x,y,c;
bool viz[200003];

int prim(int x)
{
    int i,j,cst=0,ct,nod;
    viz[x]=1;
    for(i=1; i<=n-1; i++)
    {
        for(j=0; j<v[x].size(); j++)
        {
            nod=v[x][j].sd;
            ct=v[x][j].ft;
            if(!viz[nod]) h.push({ct,{x,nod}});
        }
        while(viz[h.top().second.second]) h.pop();
        cst+=h.top().ft;
        viz[h.top().sd.sd]=1;
        sol.push_back({h.top().sd.ft,h.top().sd.sd});
        x=h.top().second.second;
        h.pop();
    }

    return cst;

}
int main()
{
    int i;
    f>>n>>m;
    for(i=1; i<=m; i++)
    {
        f>>x>>y>>c;
        v[x].pb({c,y});
        v[y].pb({c,x});
    }
    g<<prim(1)<<'\n'<<n-1<<'\n';
    for(i=0; i<sol.size(); i++)
        g<<sol[i].ft<<' '<<sol[i].sd<<'\n';
    return 0;
}