Submission details
Task:Company Queries II
Sender:aalto26gh_014
Submission time:2026-10-19 01:44:26 +0300
Language:C++ (C++20)
Status:COMPILE ERROR

Compiler report

input/code.cpp: In function 'int main()':
input/code.cpp:53:16: error: 'log2' was not declared in this scope; did you mean 'logN'?
   53 |     int logN = log2(2*n);
      |                ^~~~
      |                logN

Code

#include <vector>
#include <iostream>
using namespace std;

vector<vector<int>> ancestors;  // a[node][lb(x)]
vector<int> depth;
vector<bool> been;
vector<int> dfsOrder;
vector<int> boss;

int dfs(int node) {
    if (been[node]) return depth[node];
    been[node] = true;
    if (node == 0) return depth[node];
    int d = dfs(boss[node]) + 1;
    dfsOrder.push_back(node);
    depth[node] = d;
    return d;
}

int ancestor(int node, int up) {
    for (int ind=0; ind<20; ind++) {
        if ((1 << ind) & up) {
            node = ancestors[node][ind];
        }
    }
    return node;
}

int main() {
    int n, q;
    cin >> n >> q;
    ancestors.resize(n);
    depth.resize(n);
    been.resize(n);
    dfsOrder.resize(0);
    boss.resize(n);
    for (int i=0; i<n; i++) { been[i] = false; }

    boss[0] = 0;
    depth[0] = 0;
    dfsOrder.push_back(0);
    
    int a, b;
    for (int i=1; i<n; i++) {
        cin >> b;
        b--;
        boss[i] = b;
    }

    for (int i=0; i<n; i++) { dfs(i); }

    int logN = log2(2*n);
    
    // set ancestors
    for (int i=0; i<n; i++) { ancestors[i].push_back(boss[i]); }
    for (int i=0; i<logN; i++) {
        for (int j=0; j<n; j++) {
            ancestors[dfsOrder[j]].push_back(ancestors[ancestors[dfsOrder[j]][i]][i]);
        }
    }
    int da, db;
    for (int i=0; i<q; i++) {
        cin >> a >> b;
        a--; b--;
        da = depth[a];
        db = depth[b];
        if (da > db) swap(a, b);  // b is lower
        b = ancestor(b, abs(da - db));  // level

        // binary search
        int k = 0;
        for (int up = n/2; up >= 1; up /= 2) {
            while (k+up < n && (ancestor(a, k+up) != ancestor(b, k+up))) k += up;
        }
        cout << ancestor(a, k + 1) + 1 << endl;
    }

    return 0;
}