Cod sursa(job #1001230)

Utilizator andrettiAndretti Naiden andretti Data 24 septembrie 2013 19:17:27
Problema Radiatie Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.73 kb
#include<stdio.h>
#include<algorithm>
#define maxn 15005
#define maxm 30005
using namespace std;

struct edg{int x,y,c;} a[maxm];
int n,m,K;
int father[maxn],h[maxn],cst[maxn],lev[maxn];

void read()
{
    scanf("%d%d%d",&n,&m,&K);
    for(int i=1;i<=m;i++)
        scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].c);
}

int search(int k)
{
    if(father[k]==k) return k;
    return search(father[k]);
}

int find_lev(int k,int nr)
{
    if(father[k]==k) return nr;
    return find_lev(father[k],nr+1);
}

bool cmp(const edg &a,const edg &b){
    return a.c<b.c;
}

void kruskal()
{
    int rx,ry;

    sort(a+1,a+m+1,cmp);
    for(int i=1;i<=n;i++) father[i]=i,h[i]=1;
    for(int i=1;i<=m;i++)
    {
        rx=search(a[i].x); ry=search(a[i].y);
        if(rx!=ry)
        {
            if(h[rx]<=h[ry])
            {
                father[rx]=ry;
                cst[rx]=a[i].c;
                h[ry]+=h[rx];
            }
            else
            {
                father[ry]=rx;
                cst[ry]=a[i].c;
                h[rx]+=h[ry];
            }
        }
    }

    for(int i=1;i<=n;i++)
     if(!lev[i]) lev[i]=find_lev(i,1);
}

void solve()
{
    int x,y,sol=0,k,j;
    for(int i=1;i<=K;i++)
    {
        scanf("%d%d",&x,&y); sol=0;
        for(k=x;lev[k]>lev[y];k=father[k]) sol=max(sol,cst[k]);
        for(j=y;lev[j]>lev[x];j=father[j]) sol=max(sol,cst[j]);
        for(;k!=j;k=father[k],j=father[j]) sol=max(sol,max(cst[k],cst[j]));

        printf("%d\n",sol);
    }
}

int main()
{
    freopen("radiatie.in","r",stdin);
    freopen("radiatie.out","w",stdout);

    read();
    kruskal();
    solve();

    fclose(stdin);
    fclose(stdout);
    return 0;
}