Cod sursa(job #2351291)

Utilizator bodea.georgianaBodea Georgiana bodea.georgiana Data 22 februarie 2019 10:19:16
Problema Politie Scor 0
Compilator cpp-64 Status done
Runda Arhiva ICPC Marime 1.48 kb
#include <cstdio>
#include <algorithm>
#define N 250002
#define M 500002
#include <set>

using namespace std;
FILE *f,*g;

struct compar
{
    bool operator() (int A, int B)
        {
            return (A>B);
        }
};
set <int,compar> sol;
set <int> :: iterator it;
struct bla
{
    int x,y,c,p;
}v[M];
bool how(bla A, bla B)
{
    if(A.c!=B.c)
        return (A.c<B.c);
    return (A.p<B.p);
}
int tata[N],rang[N];
int root(int nod)
{
    while(nod!=tata[nod])
        nod=tata[nod];
    return nod;
}

void unite(int ma, int mb)
{
    if(rang[ma]<rang[mb])
        tata[ma]=mb;
    else
        tata[mb]=ma;
    if(rang[ma]==rang[mb])
        ++rang[ma];
}

int main()
{
    f=fopen("politie.in","r");
    g=fopen("politie.out","w");
    int cat,k,n,m,a,b,ma,mb;
    fscanf(f,"%d %d %d %d",&n,&m,&cat,&k);
    bla val;
    for(int i=1;i<=m;++i)
        fscanf(f,"%d %d %d %d",&v[i].x,&v[i].y,&v[i].c,&v[i].p);
    for(int i=1;i<=n;++i)
        tata[i]=i;
    int nr=1;
    sort(v+1,v+m+1,how);
    for(int i=1;i<=m;++i)
    {
        val=v[i];
        a=val.x;
        b=val.y;
        ma=root(a);
        mb=root(b);
        if(ma!=mb)
        {
            unite(ma,mb);
            sol.insert(val.p);
        }
    }
    for(it=sol.begin();it!=sol.end();++it)
        if(nr<=k)
            ++nr,fprintf(g,"%d\n",(*it));
        else
        {
            break;
        }

    fclose(f);
    fclose(g);
    return 0;
}