Cod sursa(job #2368125)

Utilizator tziplea_stefanTiplea Stefan tziplea_stefan Data 5 martie 2019 13:58:40
Problema Arbore partial de cost minim Scor 80
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.57 kb
#include <fstream>
#include <vector>
#include <set>

using namespace std;

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

const int VAL=200005;
const int INF=2000000000;

int N, M, i, j, nod, SUM, NODES;
int A, B, C, FAT[VAL], dist[VAL];
bool viz[VAL];
vector < pair <int ,int> > G[VAL];
vector < pair <int, int> > :: iterator itG;
set < pair <int, int> > Heap;
set < pair <int, int> > :: iterator it;

int main()
{
    fin >> N >> M;
    for (i=1; i<=M; i++)
    {
        fin >> A >> B >> C;
        G[A].push_back({B, C});
        G[B].push_back({A, C});
    }
    for (i=2; i<=N; i++)
    {
        dist[i]=INF;
        Heap.insert({INF, i});
    }
    Heap.insert({0, 1});
    while (1)
    {
        if (NODES==N)
            break;
        if (Heap.empty()==true)
            break;
        it=Heap.begin();
        i=it->second;
        Heap.erase(it);
        if (viz[i]==true)
            continue;
        viz[i]=true;
        NODES++;
        for (itG=G[i].begin(); itG!=G[i].end(); itG++)
        {
            nod=itG->first;
            if (dist[nod]>itG->second && viz[nod]==false)
            {
                dist[nod]=itG->second;
                FAT[nod]=i;
                Heap.insert({dist[nod], nod});
            }
        }
    }
    for (i=1; i<=N; i++)
        SUM+=dist[i];
    fout << SUM << '\n';
    fout << N-1 << '\n';
    for (i=1; i<=N; i++)
    {
        if (FAT[i]==0)
            continue;
        fout << FAT[i] << " " << i << '\n';
    }
    fin.close();
    fout.close();
    return 0;
}