Cod sursa(job #2576784)

Utilizator iliescualexiaIliescu Alexia iliescualexia Data 6 martie 2020 22:49:28
Problema Arbore partial de cost minim Scor 0
Compilator cpp-64 Status done
Runda imded Marime 1.27 kb
#include <iostream>
#include <fstream>
#include <queue>
#include <vector>
#define s second
#define f first
using namespace std;
ifstream fin("apm.in");
ofstream fout("apm.out");
vector <pair <int,int > > v[200010];
priority_queue <pair<int,int>, vector <pair <int,int > >, greater <pair <int,int > > >q;
int n,m,i,x,y,c,viz[200010],t[200010],sum;
pair<int,int> aux;
void prim(int st)
{
    viz[st]=1;
    int nrm=0,ok=0;
    while(nrm<n-1)
    {
        if(ok==0)
        {
            for(int i=0;i<v[st].size();i++)
            {
               if(viz[v[st][i].s]==0)
               {
                   q.push(v[st][i]);
               }
            }
        }
        if(viz[q.top().s]==0)
        {
            nrm++;
            t[q.top().s]=st;
            viz[q.top().s]=1;
            sum=sum+q.top().f;
            ok=0;
        }
        else
        {
            ok=1;
        }
        st=q.top().s;
        q.pop();
    }
}
int main()
{
    fin>>n>>m;
    for(i=1;i<=m;i++)
    {
        fin>>x>>y>>c;
        v[x].push_back(make_pair(c,y));
        v[y].push_back(make_pair(c,x));
    }
    prim(1);
    fout<<sum<<'\n'<<n-1<<'\n';
    for(int i=2;i<=n;i++)
    {
        fout<<i<<" "<<t[i]<<'\n';
    }
    return 0;
}