Cod sursa(job #1570482)

Utilizator VladTiberiuMihailescu Vlad Tiberiu VladTiberiu Data 16 ianuarie 2016 16:11:52
Problema Arbore partial de cost minim Scor 90
Compilator cpp Status done
Runda Arhiva educationala Marime 1.57 kb
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <vector>
#include <set>
#include <queue>
#include <stack>
#include <fstream>
#include <iostream>
#include <cstring>
#include <iomanip>

#define NMax 200005
#define INF 0x3f3f3f3f
using namespace std;
ifstream f("apm.in");
ofstream g("apm.out");

set<pair<int,int> > H;
vector<pair<int,int> > vecini[NMax];
int dist[NMax],tata[NMax],viz[NMax];
int n,m,x,y,c;

void rezolva(){
    int s = 0;
    for(int i = 1; i <= n; ++i){
        s += dist[i];
    }
    g << s <<'\n';
    g << n - 1 <<'\n';
    for(int i = 2; i <= n; i++){
        g << i << ' ' << tata[i] << '\n';
    }
}
void prim(int x0){
   // memset(tata,0,sizeof(tata));
   // memset(viz,0,sizeof(viz));
    memset(dist,INF,sizeof(dist));

    dist[x0] = 0;
    H.insert(make_pair(0,x0));

    while(!H.empty()){
        int nod = (*H.begin()).second;
        H.erase(*H.begin());
        viz[nod] = 1;
        for(int i = 0; i < vecini[nod].size(); ++i){
            if(dist[vecini[nod][i].first] > vecini[nod][i].second && !viz[vecini[nod][i].first]){
                dist[vecini[nod][i].first] = vecini[nod][i].second;
                tata[vecini[nod][i].first] = nod;
                H.insert(make_pair(dist[vecini[nod][i].first], vecini[nod][i].first));
            }
        }
    }
    rezolva();
}
int main()
{
    f >> n >> m;
    for(int i = 0; i < m; ++i){
        f >> x >> y >> c;
        vecini[x].push_back(make_pair(y,c));
        vecini[y].push_back(make_pair(x,c));
    }
    prim(1);
    return 0;
}