Cod sursa(job #2554420)

Utilizator alex.carpCarp Alexandru alex.carp Data 22 februarie 2020 21:26:03
Problema Radiatie Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.26 kb
#include <fstream>
#include <queue>
using namespace std;
ifstream f("radiatie.in");
ofstream g("radiatie.out");
struct muchie
{
    int n1,n2,cost;
};
struct comp
{
    bool operator() (muchie i,muchie j)
    {
        return i.cost>j.cost;
    }
};
vector<int>apm[15001];
priority_queue<muchie,vector<muchie>,comp>coada;
int n,m,nrm,tata[15001],rang[15001],k,costul[15001],radacinapr;
int radacina(int x)
{
    while(x!=tata[x])x=tata[x];
    return x;
}
void kruskal()
{
    for(int i=1; i<=n; i++)
    {
        tata[i]=i;
        rang[i]=1;
    }
    while(!coada.empty() && nrm<n-1)
    {
        muchie mcrt=coada.top();
        coada.pop();
        int r1=radacina(mcrt.n1);
        int r2=radacina(mcrt.n2);
        if(r1!=r2)
        {
            if(rang[r1]<rang[r2])
            {
                tata[r1]=r2;
                radacinapr=r2;
                apm[r2].push_back(r1);
                costul[r1]=mcrt.cost;
            }
            else
            {
                tata[r2]=r1;
                radacinapr=r1;
                apm[r1].push_back(r2);
                costul[r2]=mcrt.cost;

            }
            if(rang[r1]==rang[r2])
                rang[r1]++;
            nrm++;
        }
    }
}
void dfs(int x,int tatal)
{
    rang[x]=rang[tatal]+1;
    for(int i=0; i<apm[x].size(); i++)
    {
            dfs(apm[x][i],x);
    }

}
int main()
{
    f>>n>>m>>k;
    for(int i=1; i<=m; i++)
    {
        int a,b,c;
        f>>a>>b>>c;
        muchie x;
        x.n1=a;
        x.n2=b;
        x.cost=c;
        coada.push(x);
    }
    kruskal();
    for(int i=1; i<=n; i++)
    {
        rang[i]=0;
    }
    dfs(radacinapr,radacinapr);
    for(int i=1; i<=k; i++)
    {
        int x,y;
        int maxim=0;
        f>>x>>y;
        while(rang[x]<rang[y])
        {
            maxim=max(maxim,costul[y]);
            y=tata[y];
        }
        while(rang[x]>rang[y])
        {
            maxim=max(maxim,costul[x]);
            x=tata[x];
        }
        while(x!=y)
        {
            maxim=max(maxim,costul[x]);
            maxim=max(maxim,costul[y]);
            y=tata[y];
            x=tata[x];
        }
        g<<maxim<<'\n';
    }
    return 0;
}