Cod sursa(job #3193683)

Utilizator Theo20067Cismaru Theodor-Alexe Theo20067 Data 15 ianuarie 2024 12:55:37
Problema Radiatie Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.12 kb
#include <fstream>
#include <algorithm>
#include <vector>
#include <bitset>
using namespace std;
ifstream fin ("radiatie.in");
ofstream fout("radiatie.out");
int n,m,Q,i,j,nr,T[15003],D[15003],nivel[15003],r,x,y,maxim;
struct elem
{
    int x,y,cost;
}V[40003];
vector < pair<int,int> > A[40003];
bool cmp(elem a,elem b){ return a.cost<b.cost;}
int get_root(int x)
{
    while(T[x]>0)
        x=T[x];
    return x;
}
void join(int x,int y)
{
    int root_a=get_root(x);
    int root_b=get_root(y);
    if(root_a==root_b)
        return;
    if(T[root_a]>T[root_b])
        swap(root_a,root_b);
    T[root_a]+=T[root_b];
    T[root_b]=root_a;
}
bool query(int x, int y){ return get_root(x)!=get_root(y);}
void dfs(int x,int niv)
{
    nivel[x]=niv;
    for(int j=0;j<A[x].size();j++)
    {
        int vecin=A[x][j].first;
        int cost=A[x][j].second;
        if(!nivel[vecin])
        {
            D[vecin]=cost;
            T[vecin]=x;
            dfs(vecin,niv+1);
        }
    }
}
void solve(int x,int y)
{
    while(nivel[x]>nivel[y])
    {
        maxim=max(maxim,D[x]);
        x=T[x];
    }
    while(nivel[y]>nivel[x])
    {
        maxim=max(maxim,D[y]);
        y=T[y];
    }
    while(x!=y)
    {
        maxim=max(maxim,max(D[x],D[y]));
        x=T[x];
        y=T[y];
    }
}
int main()
{
    fin>>n>>m>>Q;
    for(i=1;i<=m;i++)
        fin>>V[i].x>>V[i].y>>V[i].cost;
    for(i=1;i<=n;i++)
        T[i]=-1;
    sort(V+1,V+m+1,cmp);
    for(i=1;i<=m;i++)
    {
        if(query(V[i].x,V[i].y))
        {
            A[get_root(V[i].x)].push_back({get_root(V[i].y),V[i].cost});
            A[get_root(V[i].y)].push_back({get_root(V[i].x),V[i].cost});

            join(V[i].x,V[i].y);
            nr++;
            if(nr==n-1)
                break;
        }
    }
    for(i=1;i<=n;i++)
        if(T[i]<0)
        {
            r=i;
            break;
        }
    dfs(r,1);
    T[r]=0;
    D[r]=0;
    for(i=1;i<=Q;i++)
    {
        fin>>x>>y;
        maxim=-1000000001;
        solve(x,y);
        fout<<maxim<<"\n";
    }
    return 0;
}