Cod sursa(job #2468944)

Utilizator mihaimodiMihai Modi mihaimodi Data 6 octombrie 2019 11:58:12
Problema Arbore partial de cost minim Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.24 kb
#include <fstream>
#include <vector>
#include <queue>
#define inf 2147483647
using namespace std;
ifstream fin("apm.in");
ofstream fout("apm.out");
int d[400001],tata[400001];
int n,m,x,y,c,poz,Min,s;
bool viz[400001];
struct cmp
{
    bool operator ()(const int &a, const int &b)
    {
        return d[a]>d[b];
    }
};
vector <pair <int, int > > g[400001];
priority_queue < int , vector <int>  ,cmp > h;
int main()
{
    fin>>n>>m;
    for(int i=1;i<=m;i++)
    {
        fin>>x>>y>>c;
        g[x].push_back(make_pair(y,c));
        g[y].push_back(make_pair(x,c));
    }
    for(int i=2;i<=n;i++)
        d[i]=inf;
    for(int i=1;i<=n;i++)
    {
        Min=inf,poz=0;
        for(int j=1;j<=n;j++)
            if(viz[j]==0&&d[j]<Min)
            {
                Min=d[j];
                poz=j;
            }
        viz[poz]=1;
        s+=d[poz];
        for(int j=0;j<g[poz].size();j++)
            if(viz[g[poz][j].first]==0&&g[poz][j].second<d[g[poz][j].first])
            {
                d[g[poz][j].first]=g[poz][j].second;
                tata[g[poz][j].first]=poz;
            }
    }
    fout<<s<<'\n'<<n-1<<'\n';
    for(int i=2;i<=n;i++)
        fout<<tata[i]<<' '<<i<<'\n';
    return 0;
}