Cod sursa(job #947750)

Utilizator xxxcnmvxxxnume cpmplet xxxcnmvxxx Data 8 mai 2013 12:52:29
Problema Radiatie Scor 60
Compilator cpp Status done
Runda Arhiva de probleme Marime 3.83 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <algorithm>

using namespace std;

ifstream in ("radiatie.in");
ofstream out ("radiatie.out");

const int MAXN = 15010;

struct muchie
{
    int a, b, c;
} X[MAXN * 2];

struct query
{
    int x, y;
} Q[MAXN];

int N, M, K;
vector < pair <int, int> > Graf[MAXN];
int T[MAXN];
int H[MAXN * 4], First[MAXN], Lvl[MAXN * 4];
int Log[MAXN], RMQ[16][MAXN * 2];
int D[16][MAXN], Smen[16][MAXN];
bool Intra[MAXN], Viz[MAXN];
int now;

int find (int nod)
{
    if (T[nod] == nod)
        return nod;

    return (T[nod] = find (T[nod]));
}

struct comp
{
    inline bool operator () (const muchie &A, const muchie &B) const
    {
        return A.c < B.c;
    }
};

void APM ()
{
    int i, x, y, c, Tx, Ty;

    sort (X + 1, X + M + 1, comp ());

    for (i = 1; i <= M; i ++)
        T[i] = i;

    for (i = 1; i <= M; i ++){
        x = X[i].a;
        y = X[i].b;
        c = X[i].c;
        Tx = find (x);
        Ty = find (y);

        if (Tx != Ty){
            T[Tx] = Ty;
            Graf[x].push_back (make_pair (y, c));
            Graf[y].push_back (make_pair (x, c));
            Intra[y] = 1;
        }
    }
}

void DFS (pair <int, int> nod, int father, int lev)
{
    now ++;
    H[now] = nod.first;
    Lvl[now] = lev;
    First[nod.first] = now;
    D[0][nod.first] = father;
    Smen[0][nod.first] = nod.second;

    for (vector < pair <int, int> > :: iterator it = Graf[nod.first].begin (); it != Graf[nod.first].end (); ++ it){
        if (Viz[it -> first])
            continue;

        Viz[it -> first] = 1;
        DFS (*it, nod.first, lev + 1);
        ++ now;
        H[now] = nod.first;
        Lvl[now] = lev;
    }
}

void Read ()
{
    int i;

    in >> N >> M >> K;

    for (i = 1; i <= M; i ++)
        in >> X[i].a >> X[i].b >> X[i].c;

    for (i = 1; i <= K; i ++)
        in >> Q[i].x >> Q[i].y;
}

void Build_Log ()
{
    int i;

    for (i = 2; i < MAXN; i ++)
        Log[i] = Log[i / 2] + 1;
}

void Build_RMQ ()
{
    int i, j, a, b;

    for (i = 1; i <= now; i ++)
        RMQ[0][i] = i;

    for (j = 1; (1 << j) <= now; j ++)
        for (i = 1; i + (1 << j) <= now; i ++){
            a = RMQ[j - 1][i];
            b = RMQ[j - 1][i + (1 << (j - 1))];

            if (Lvl[a] <= Lvl[b])
                RMQ[j][i] = a;
            else
                RMQ[j][i] = b;
        }
}

int LCA (int nod1, int nod2)
{
    nod1 = First[nod1];
    nod2 = First[nod2];

    if (nod2 < nod1)
        swap (nod1, nod2);

    int dif = nod2 - nod1 + 1;
    int log = Log[dif];
    int a, b;
    a = RMQ[log][nod1];
    b = RMQ[log][nod2 - (1 << log) + 1];

    if (Lvl[a] <= Lvl[b])
        return a;
    else
        return b;
}

void Build_Smen ()
{
    int i, j;

    for (i = 1; (1 << i) <= N; i ++)
        for (j = 1; j <= N; j ++){
            D[i][j] = D[i - 1][D[i - 1][j]];
            Smen[i][j] = max (Smen[i - 1][j], Smen[i - 1][D[i - 1][j]]);
        }
}

int Query (int x, int y)
{
    int i, dif = Lvl[First[x]] - Lvl[First[y]], Ans = -1;

    for (i = 0; (1 << i) <= dif; i ++){
        if (dif & (1 << i)){
            if (Smen[i][x] > Ans)
                Ans = Smen[i][x];
            x = D[i][x];
        }
    }

    return Ans;
}

int Solve (int x, int y)
{
    int lca = H[LCA (x, y)];
    int a, b;

    a = Query (x, lca);
    b = Query (y, lca);

    return max (a, b);
}

void Init ()
{
    APM ();
    Viz[1] = 1;
    DFS (make_pair (1, 0), 0, 1);
    Build_Log ();
    Build_RMQ ();
    Build_Smen ();
}

void Solve ()
{
    int i;

    for (i = 1; i <= K; i ++)
        out << Solve (Q[i].x, Q[i].y) << "\n";
}

int main()
{
    Read ();
    Init ();
    Solve ();

    return 0;
}