Cod sursa(job #1281109)

Utilizator margikiMargeloiu Andrei margiki Data 2 decembrie 2014 20:53:46
Problema Arbore partial de cost minim Scor 20
Compilator cpp Status done
Runda Arhiva educationala Marime 1.14 kb
# include <fstream>
# include <vector>
# include <algorithm>
using namespace std;
ifstream f("apm.in");
ofstream g("apm.out");
struct muchie
{
    int x, y, cost;
}a[400005];
bool cmp (muchie x, muchie y)
{
    if (x.cost>=y.cost) return 0;
       else return 1;
}
int i,j,n,m,x,y,VV,luate,costfinal;
int R[200005],marime[200005],sol[200005];
int radacina (int x)
{
    while (x!=R[x])
        x=R[x];
    return x;
}
void APM ()
{
    int i,Rx,Ry;
    luate=1;
    for (i=1; i<=m && luate!=n; ++i)
    {
        Rx=radacina(a[i].x);
        Ry=radacina(a[i].y);
        if (Rx!=Ry)
        {
            ++luate; costfinal+=a[i].cost; sol[++VV]=i;
            if (marime[Rx]>marime[Ry]) R[a[i].y]=Rx, marime[a[i].x]+=marime[a[i].y];
            else R[a[i].x]=Ry, marime[a[i].y]+=marime[a[i].x];
        }
    }
}
int main ()
{
    f>>n>>m;
    for (i=1; i<=m; ++i)
    {
        if (i<=n) R[i]=i, marime[i]=1;
        f>>a[i].x>>a[i].y>>a[i].cost;
    }
    sort (a+1,a+m+1,cmp);
    APM ();
    g<<costfinal<<"\n"<<VV<<"\n";
    for (i=1; i<=VV; ++i)
        g<<a[sol[i]].x<<" "<<a[sol[i]].y<<"\n";
    return 0;
}