Cod sursa(job #2092619)

Utilizator giotoPopescu Ioan gioto Data 21 decembrie 2017 23:37:38
Problema Radiatie Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.11 kb
#include <bits/stdc++.h>
using namespace std;

int n, m, k, x, y;
int TT[15005], RG[15005], H[15005], T2[15005], cost[15005], C[15005], C2[15005];
bool f[15005];
const int h = 205;
vector <pair <int, int> > v[15005];
struct V{
    int x, y, c;
    bool operator < (const V &aux)const{
        return c < aux.c;
    }
};
V a[30005];
inline int find(int x){
    int R;
    for(R = TT[x]; TT[R] != R ; R = TT[R]) ;
    while(x != TT[x]){
        int y = TT[x];
        TT[x] = R;
        x = y;
    }
    return TT[x];
}
inline void unire(int x, int y){
    if(RG[x] > RG[y]) TT[y] = x;
    else if(RG[y] < RG[x]) TT[x] = y;
    else{
        ++RG[y];
        TT[x] = y;
    }
}
inline void dfs(int nod, int t2, int lev, int c2){
    T2[nod] = t2;
    C2[nod] = c2;
    H[nod] = lev;
    if(lev % h == 0) t2 = nod, c2 = 0;
    for(auto it : v[nod]){
        if(TT[nod] == it.first) continue ;
        TT[it.first] = nod;
        C[it.first] = it.second;
        dfs(it.first, t2, lev + 1, max(c2, it.second));
    }
}
inline int lca(int x, int y){
    int Sol = 0;
    while(T2[x] != T2[y]){
        if(H[x] > H[y]) Sol = max(C2[x], Sol), x = T2[x];
        else Sol = max(C2[y], Sol), y = T2[y];
    }
    while(x != y){
        if(H[x] > H[y]) Sol = max(C[x], Sol), x = TT[x];
        else Sol = max(C[y], Sol), y = TT[y];
    }
    return Sol;
}
int main()
{
    freopen("radiatie.in", "r", stdin);
    freopen("radiatie.out", "w", stdout);
    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);
    for(int i = 1;  i <= n ; ++i){
        TT[i] = i; RG[i] = 1;
    }
    sort(a + 1, a + m + 1);
    for(int i = 1; i <= m ; ++i){
        if(find(a[i].x) != find(a[i].y)){
            unire(find(a[i].x), find(a[i].y));
            v[a[i].x].push_back(make_pair(a[i].y, a[i].c));
            v[a[i].y].push_back(make_pair(a[i].x, a[i].c));
        }
    }
    TT[1] = 1;
    dfs(1, 0, 0, 0);
    for(int i = 1; i <= k ; ++i){
        scanf("%d%d", &x, &y);
        printf("%d\n", lca(x, y));
    }

    return 0;
}