Cod sursa(job #3030117)

Utilizator NiffSniffCojocaru Calin Marcu NiffSniff Data 17 martie 2023 15:25:49
Problema Arbore partial de cost minim Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.36 kb
#include <fstream>
#include <vector>
#include <queue>
#include <bitset>
using namespace std;
string file = "apm";
ifstream cin (file + ".in");
ofstream cout (file + ".out");
const int N = 2e5,INF = 1001;
struct muchie{
    int vf;
    int cost;
    bool operator < (const muchie &y) const
    {
        return cost > y.cost;
    }
};

vector <muchie> a[N+1];
bitset <N+1> in_t;
int n, d[N+1],predecesor[N+1];
int prim()
{
    priority_queue <muchie> h;
    h.push({1,0});
    int cost_total = 0;
    d[1] = 0;
    for (int i=2; i<=n; i++)
    {
        d[i] = INF;
    }
    while (!h.empty())
    {
        int x = h.top().vf;
        h.pop();
        if (in_t[x])
        {
            continue;
        }
        cost_total += d[x];
        in_t[x] = 1;
        for (auto e : a[x])
        {
            int c = e.cost, y = e.vf;
            if (c < d[y])
            {
                d[y] = c;
                predecesor[y] = x;
                h.push({y,c});
            }
        }
    }
    return cost_total;
}

int main ()
{
    int m,x,y,z;
    cin >> n >> m;
    while (m--)
    {
        cin >> x >> y >> z;
        a[x].push_back({y,z});
        a[y].push_back({x,z});
    }
    cout << prim() << '\n' << n-1;
    for (int i=1; i<=n; i++)
    {
        cout << '\n' << i << ' ' << predecesor[i];
    }
}