Cod sursa(job #793714)

Utilizator lucian666Vasilut Lucian lucian666 Data 3 octombrie 2012 21:20:26
Problema Lazy Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.35 kb


//Vasilut
#include<fstream>
#include<algorithm>

#define NN 200001
using namespace std;
ofstream out("lazy.out");

struct drumusor{
    int x,y;
    long long c1,c2;
    };
    drumusor v[NN];

int T[NN],n,m,poz[NN],k;

bool cmp(drumusor i,drumusor j)
{
    return i.c1 < j.c1 || ( i.c1 == j.c1  && i.c2 > j.c2 ) ;
}

int cauta(int );
void unire(int ,int );
void init();
void read();
void kruskal();
void write();

int main()
{
    read();
    kruskal();
    write();
    return 0;
}

void read()
{
    ifstream in("lazy.in");
    in>>n>>m;
    int k=0;
    for(int x,y,i=1;i<=m;i++)
    {

        long long c1,c2;
        in>>x>>y>>c1>>c2;
        v[i].x=x;
        v[i].y=y;
        v[i].c1=c1;
        v[i].c2=c2;
    }
}

void init()
{
    for(int i=1;i<=n;i++)
        T[i]=i;
}

int cauta(int nod)
{
    if( nod !=T[nod] )
    T[nod]=cauta(T[nod]);
        return T[nod];
}

void unire(int nod1,int nod2)
{
    T[cauta(nod1)]=cauta(nod2);
}

void kruskal()
{
    init();
    sort(v+1,v+m+1,cmp);
    for(int i=1;i<=m;i++)
        if( cauta(v[i].x ) != cauta(v[i].y) )
        {
            ++k;
            poz[k]=i;
            unire(v[i].x,v[i].y);
        }
        if ( k == n-1 )
            return ;
}

void write()
{
    for(int i=1;i<=k;++i)
    out<<poz[i]<<'\n';
}