Cod sursa(job #2673404)

Utilizator mex7Alexandru Valentin mex7 Data 16 noiembrie 2020 18:41:16
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.49 kb
#include <bits/stdc++.h>
#define ll long long
using namespace std;
ifstream fin("apm.in");
ofstream fout("apm.out");
vector <pair <int, ll> > adj[200001];
bitset <200001> reached;
pair <ll, int> minCost[200001]; //.first == cost; .second == root;

void dijkstra(int start) {
    priority_queue <pair <ll, int> > best;
    best.push(make_pair(0, start));

    while (!best.empty()) {
        int front = best.top().second;
        ll cost = -best.top().first;
        best.pop();

        reached[front] = 1;
        for (pair <int, ll> edge : adj[front])
            if (!reached[edge.first])
                if (edge.second <= minCost[edge.first].first) {
                    minCost[edge.first].first = edge.second;
                    minCost[edge.first].second = front;
                    best.push(make_pair(-edge.second, edge.first));
                }
    }
}

void addEdge(int x, int y, int c) {
    adj[x].push_back(make_pair(y, c));
    adj[y].push_back(make_pair(x, c));
}

int main() {
    int n, m, x, y, c;

    fin >> n >> m;
    for (int i = 1; i <= m; i++) {
        fin >> x >> y >> c;
        addEdge(x, y, c);
    }

    for (int i = 2; i <= n; i++)
        minCost[i].first = INT_MAX;

    dijkstra(1);

    ll result = 0;
    for (int i = 1; i <= n; i++)
        result += minCost[i].first;

    fout << result << '\n' << n - 1 << '\n';
    for (int i = 2; i <= n; i++)
        fout << i << ' ' << minCost[i].second << '\n';

    return 0;
}