Cod sursa(job #2545845)

Utilizator sichetpaulSichet Paul sichetpaul Data 13 februarie 2020 16:33:17
Problema Lowest Common Ancestor Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.73 kb
#include <bits/stdc++.h>
#define Nmax 32005
using namespace std;

ifstream f("concurs.in");
ofstream g("concurs.out");

int N, M, K;
vector <int> G[Nmax];
int E[2 * Nmax], lg[2 * Nmax], lev[2 * Nmax], rmq[2 * Nmax][20], pos[Nmax], v[Nmax];

void update(int node, int father) {
    E[++K] = node;
    lev[K] = lev[pos[father]] + 1;
    if (!pos[node]) pos[node] = K;
}
void DFS(int node, int father) {
    update(node, father);
    for (auto it: G[node]) {
        DFS(it, node);
        update(node, father);
    }
}
void RMQ() {
   for (int i = 2; i <= K; ++i)
      lg[i] = lg[i / 2] + 1;
   for (int i = 1; i <= K; ++i)
      rmq[i][0] = i;

   for (int j = 1; (1 << j) <= K; ++j)
   for (int i = 1; i + (1 << j) - 1 <= K; ++i) {
      int a = rmq[i][j - 1], b = rmq[i + (1 << (j - 1))][j - 1];
      if (lev[a] < lev[b]) rmq[i][j] = a;
      else rmq[i][j] = b;
   }
}
int compute(int x, int y) {
   int mid = lg[y - x + 1];
   int a = rmq[x][mid], b = rmq[y - (1 << mid) + 1][mid];
   if (lev[a] < lev[b]) return E[a];
   return E[b];
}
int main()
{
    f >> N >> M;
    for (int i = 1; i <= N; ++i)
        f >> v[i];
    for (int i = 1; i < N; ++i) {
        int x, y;
        f >> x >> y;
        G[x].push_back(y);
    }
    DFS(1, 0), RMQ();

    int ans = 0, a, b;
    for (int i = 1; i <= M; ++i) {
        int x, y; bool s = 0;
        f >> x >> y;
        if (pos[x] > pos[y]) swap(x, y), s = 1;
        int lca = compute(pos[x], pos[y]);
        if (s) swap(x, y);

        if (v[lca] > ans) ans = v[lca], a = x, b = y;
        else if (v[lca] == ans)
           if (x < a || (x == a && y < b)) a = x, b = y;
    }
        g << ans << " " << a << " " << b << '\n';

    return 0;
}