Cod sursa(job #2657204)

Utilizator WholeGrainAndy N WholeGrain Data 10 octombrie 2020 07:31:46
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.68 kb
#include <iostream>
#include <vector>
#include <set>
#include <set>
#include <fstream>

using namespace std;

#define CAP 200001

#define INFIN 1e9

struct Edge {
    int node;

    int cost;

    bool operator < (Edge e) const {
        if (cost != e.cost) {
            return cost < e.cost;
        }

        return node < e.node;
    }
};

int n, m;

vector<Edge> g[CAP];

set<Edge> available;

bool visited[CAP];

int dist[CAP];

int sol = 0;

int solEdge[CAP];

int main() {
    ofstream out("apm.out");

    ifstream in("apm.in");

    in >> n >> m;

    for (int i = 0; i < m; i++) {
        int x, y, c;

        in >> x >> y >> c;

        x--; y--;

        g[x].push_back({y, c});

        g[y].push_back({x, c});
    }

    for (int i = 0; i < CAP; i++) {
        dist[i] = INFIN;
    }

    dist[0] = 0;

    available.insert({0, 0});

    while (!available.empty()) {
        Edge curr = *available.begin();

        available.erase(available.begin());

        visited[curr.node] = true;

        int bestEdge = -1;

        for (int i = 0; i < g[curr.node].size(); i++) {
            if (g[curr.node][i].cost < dist[g[curr.node][i].node] && !visited[g[curr.node][i].node]) {
                dist[g[curr.node][i].node] = g[curr.node][i].cost;

                available.insert({g[curr.node][i].node, dist[g[curr.node][i].node]});

                bestEdge = g[curr.node][i].node;

                solEdge[g[curr.node][i].node] = curr.node;
            }
        }
    }

    for (int i = 0; i < n; i++) {
        sol += dist[i];
    }

    out << sol << '\n';

    out << n - 1 << '\n';

    for (int i = 1; i < n; i++) {
        out << solEdge[i] + 1 << " " << i + 1 << '\n';
    }

    return 0;
}