Cod sursa(job #2126154)

Utilizator B_RazvanBaboiu Razvan B_Razvan Data 9 februarie 2018 11:42:04
Problema Arbore partial de cost minim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.45 kb
#include <iostream>
#include <cstdio>
#include <vector>
#include <queue>
#define INF 0x3f3f3f
#define NMAX 200005

using namespace std;

int N, M, dist[NMAX], tata[NMAX];
bool viz[NMAX];
vector <pair<int, int> > G[NMAX];
vector <pair<int, int> >::iterator it;
priority_queue <pair <int, int> > Q;

void read()
{
    scanf("%d %d", &N, &M);
    for(int i=1; i<=M; ++i)
    {
        int a, b, c;
        scanf("%d %d %d", &a, &b, &c);
        G[a].push_back(make_pair(b, c));
        G[b].push_back(make_pair(a, c));
    }
    for(int i=2; i<=N; ++i)
        dist[i] = INF;
}

void apmPrim()
{
    Q.push(make_pair(0, 1));
    while(!Q.empty())
    {
        int nod = Q.top().second;
        Q.pop();
        viz[nod] = true;
        for(it = G[nod].begin(); it != G[nod].end(); ++it)
            if(viz[it->first] == false && dist[it->first] > it->second)
            {
                dist[it->first] = it->second;
                tata[it->first] = nod;
                Q.push(make_pair(-1 * dist[it->first], it->first));
            }
    }
}

void print()
{
    int distTotala = 0;
    for(int i = 2; i<=N; ++i)
        distTotala += dist[i];
    printf("%d\n", distTotala);
    printf("%d\n", N-1);
    for(int i=2; i<=N; ++i)
        printf("%d %d\n", tata[i], i);
}

int main()
{
    freopen("apm.in", "r", stdin);
    freopen("apm.out", "w", stdout);
    read();
    apmPrim();
    print();
    return 0;
}