Cod sursa(job #2680272)

Utilizator PoseidonGeminiPoseidonGemini PoseidonGemini Data 3 decembrie 2020 09:11:19
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.67 kb
#include <fstream>
#include <queue>
#include <vector>
#define NM 200001
#define INF 1001

using namespace std;

ifstream fin("apm.in");
ofstream fout("apm.out");

struct pereche {
    int y, c;
};

struct muchie {
    int x, y, c;
};

struct comp {
    bool operator() (muchie u, muchie v) {
        return u.c > v.c;
    }
};

int viz[NM], t[NM], d[NM];
int n, m, x, y, cost, nr, dm, im, ct, i, j;

vector<pereche> G[NM];
priority_queue<muchie, vector<muchie>, comp> q;


int main() {
    fin >> n >> m;

    for (int i = 1; i <= m; i++) {
        fin >> x >> y >> cost;
        G[x].push_back({ y, cost });
        G[y].push_back({ x, cost });
    }

    for (int i = 2; i <= n; ++i) {
        d[i] = INF;
        t[i] = 1;
    }
    
    for (int i = 0; i < G[1].size(); i++) {
        pereche pt = G[1][i];
        d[pt.y] = pt.c;
        q.push({ 1, pt.y, pt.c });
    }

    viz[1] = 1;
    nr = 1;
    ct = 0;
    
    while (nr < n)
    {
        muchie e;

        e = q.top(); q.pop();

        if (viz[e.y] == 0) {
            dm = e.c;

            // il alegem pe im
            im = e.y;

            nr++;
            viz[im] = 1;
            ct = ct + dm;

            pereche pt;
            for (int i = 0; i < G[im].size(); ++i) {
                pt = G[im][i];
                if (viz[pt.y] == 0 && d[pt.y] > pt.c) {
                    d[pt.y] = pt.c;
                    t[pt.y] = im;
                    q.push({ im, pt.y, pt.c });
                }
            }
        }
    }

    fout << ct << '\n';
    fout << n - 1 << '\n';
    for (i = 2; i <= n; i++)
        fout << i << ' ' << t[i] << '\n';
}