Company Queries II

input/code.cpp: In function 'int main()':
input/code.cpp:15:19: warning: comparison of integer expressions of different signedness: 'int' and 'ull' {aka 'long long unsigned int'} [-Wsign-compare]
   15 |     for(int i=0; i<n-1; i++){
      |                  ~^~~~
input/code.cpp:37:19: warning: comparison of integer expressions of different signedness: 'int' and 'ull' {aka 'long long unsigned int'} [-Wsign-compare]
   37 |     for(int i=0; i<q; i++){
      |                  ~^~

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
 
int d[200001];
 
int main(){
    ull n, q;
    cin >> n >> q;
 
    map<int,vector<int>> e;
    map<int,int> boss;
 
    for(int i=0; i<n-1; i++){
        int b;
        cin >> b;
        e[b].push_back(i+2);
        boss[i+2] = b;
    }
 
    d[1] = 0;
    stack<int> s;
    s.push(1);
    while(!s.empty()){
        stack<int> s2;
        while(!s.empty()){
            int sa = s.top(); s.pop();
            for(int e : e[sa]){
                d[e] = d[sa] + 1;
                s2.push(e);
            }
        }
        s = s2;
    }
 
    for(int i=0; i<q; i++){
        int a,b;
        cin >> a >> b;
 
        int da = d[a];
        int db = d[b];
        if(da > db){
            swap(da, db);
            swap(a,b);
        }
 
        for(; da < db; db--){
            b = boss[b];
        }
 
        while(b != a){
            b = boss[b];
            a = boss[a];
        }
        cout << b << endl;
    }
}
 

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: TIME LIMIT EXCEEDED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: TIME LIMIT EXCEEDED

Test 9

Verdict: TIME LIMIT EXCEEDED

Test 10

Verdict: TIME LIMIT EXCEEDED

Test 11

Verdict: ACCEPTED

Test 12

Verdict: TIME LIMIT EXCEEDED