Cod sursa(job #2039830)

Utilizator LauraNaduLaura Nadu LauraNadu Data 14 octombrie 2017 23:31:50
Problema Lazy Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.12 kb
#include <fstream>
#include <algorithm>
#define cost1 first.first
#define cost2 first.second
#define st second.first
#define dr second.second.first
#define poz second.second.second
using namespace std;
ifstream f("lazy.in");
ofstream g("lazy.out");
int n, m, rx, ry, t[400001], s, k, p[400001], u[400001];
pair<pair<int, int>, pair<int, pair<int, int> > > l[400010];
int rad (int x){
    while(t[x]>0)
        x=t[x];
    return x;
}
int main()
{
    f>>n>>m;
    for(int i=1;i<=n;i++)
        t[i]=-1;
    for(int i=1;i<=m;i++)
    {
        f>>l[i].st>>l[i].dr>>l[i].cost1>>l[i].cost2;
        l[i].poz=i;
        l[i].cost2*=-1;
    }
    sort(l+1,l+m+1);
    for(int i=1;i<=m;i++)
    {
        rx=rad(l[i].st);
        ry=rad(l[i].dr);
        if(rx!=ry)
        {
            p[++k]=l[i].poz;
            if(t[rx]<t[ry])
            {
                t[rx]+=t[ry];
                t[ry]=rx;
            }
            else
            {
                t[ry]+=t[rx];
                t[rx]=ry;
            }
        }
    }
    for(int i=1;i<=k;i++)
        g<<p[i]<<"\n";
    return 0;
}