Cod sursa(job #2666006)

Utilizator Xutzu358Ignat Alex Xutzu358 Data 31 octombrie 2020 17:14:45
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.44 kb
#include <bits/stdc++.h>

using namespace std;

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

set < pair < int , int > > s;
vector < pair < int , int > > v[200005];
int totaldist,n,viz[200005],dist[200005],tata[200005];
int m,i,x,y,cost;
int oo=2000000009;

int main()
{
    f >> n >> m;
    for (i=1;i<=m;i++) {
        f >> x >> y >> cost;
        v[x].push_back(make_pair(y,cost));
        v[y].push_back(make_pair(x,cost));
    }
     for(int i=1;i<=n;i++) {
            dist[i]=oo;
    }
    dist[1]=0;
    s.insert(make_pair(0,1));
        while (s.empty()==0) {
            int nod = s.begin()->second;
            int distnod = s.begin()->first;
            s.erase(s.begin());
            if (viz[nod]==0) {
                viz[nod] = 1;
                totaldist += distnod;
                for (int i=0;i<v[nod].size();i++) {
                    int nodtoviz = v[nod][i].first;
                    int disttoviz = v[nod][i].second;
                    if (dist[nodtoviz] > disttoviz && viz[nodtoviz]==0) {
                        dist[nodtoviz] = disttoviz;
                        s.insert(make_pair(dist[nodtoviz],nodtoviz));
                        tata[nodtoviz] = nod;
                    }
                }
            }
    }
    g << totaldist << '\n';
    g << n-1 << '\n';
    for (i=1;i<=n;i++) {
        if (tata[i]!=0) {
            g << i << " " << tata[i] << '\n';
        }
    }
    return 0;
}