Cod sursa(job #2532408)

Utilizator CarlaDianaCarla Diana CarlaDiana Data 27 ianuarie 2020 20:04:23
Problema Arbore partial de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.86 kb

#include    <fstream>
#include    <iostream>
#include    <algorithm>

using namespace std;

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

const int NMax = 400005;

pair <int,int> P[NMax];

int N, M, Total, TT[NMax], k, RG[NMax];

struct Edge
{
    int x,y,c;

}V[NMax];

bool Compare(Edge a, Edge b)
{
    return a.c < b.c;
}

void Read()
{
    fin >> N >> M;

    for(int i = 1; i <= M; i++)
        fin >> V[i].x >> V[i].y >> V[i].c;

    sort(V+1, V+M+1, Compare);

    for(int i = 1; i <= M; i++)
        cout << V[i].x << " " << V[i].y << " " << V[i].c << "\n";
}

int Find(int nod)
{
    while(TT[nod] != nod)
        nod = TT[nod];
    return nod;
}

void Unite(int x, int y)
{
    if(RG[x] < RG[y])
        TT[x] = y;
    if(RG[y] < RG[x])
        TT[y] = x;
    if(RG[x] == RG[y])
    {
        TT[x]=y;
        RG[y]++;
    }
}

void Solve()
{
    for(int i=1;i<=M;i++)
    {
        //cout << "Incerc " << V[i].x << " cu " << V[i].y << "\n";
        if(Find(V[i].x) != Find(V[i].y))
        {
            //cout << "Unesc " << V[i].x << " cu " << V[i].y << "\n\n";
            Unite(Find(V[i].x), Find(V[i].y));
            P[++k].first    =   V[i].x;
            P[k].second     =   V[i].y;
            Total           +=  V[i].c;
        }
    }
}

int main()
{
    Read();

    for(int i = 1; i <= M; i++)
    {
        TT[i]=i;
        RG[i]=1;
    }

    Solve();

    fout << Total << "\n";
    fout << N-1 <<"\n";

    for(int i = 1; i <= k; i++)
        fout << P[i].first << " " << P[i].second << "\n";

    return 0;
}

/*
#include <fstream>
#include <algorithm>
using namespace std;
ifstream fin ("apm.in");
ofstream fout ("apm.out");
int n,m,tt[400010],xx,yy,rg[40010],q;
long long cost;
struct muchie{int x,y,c;}v[400010];
struct nu{int x,y;} k[400010];
bool comp(muchie a,muchie b)
{
    return a.c<b.c;
}
int tata(int nod)
{
    while(tt[nod]!=nod)
        nod=tt[nod];

    return nod;
}
int main()
{
    fin>>n>>m;
    for(int i=1;i<=m;i++)
    {
        fin>>v[i].x>>v[i].y>>v[i].c;
    }
    sort(v+1,v+m+1,comp);
    for(int i=1;i<=n;i++)
    {
        tt[i]=i;
        rg[i]=1;
    }


    for(int i=1;i<=m;i++)
    {
        xx=tata(v[i].x);
        yy=tata(v[i].y);
        if(xx!=yy)
        {
            cost+=v[i].c;
            if(rg[v[i].x]<rg[v[i].y])
                tt[xx]=tt[yy];
            else
            {
                if(rg[v[i].x]>rg[v[i].y])
                    tt[yy]=tt[xx];
                else
                {
                    tt[xx]=tt[yy];rg[xx]++;
                }
            }
            k[++q].x=v[i].x;
            k[q].y=v[i].y;
        }
    }
    fout<<cost<<'\n'<<n-1<<'\n';
    for(int i=1;i<=q;i++)
    {
        fout<<k[i].x<<" "<<k[i].y<<'\n';
    }
    return 0;
}
*/