Company Queries II

from math import ceil, log2

def printParentMap(parent, empCount, logHeight):
    for hh in range(logHeight):
        for ee in range(empCount):
            print(f"{parent[hh*empCount+ee]:2}", end=" ")
        print()
def run(q0, q1, bosses, heights):
    if q0 == q1:
        return q0
    if heights[q0] == heights[q1]:
        if bosses[q0] == bosses[q1]:
            return bosses[q0]
        else:
            return run(bosses[q0], bosses[q1], bosses, heights)
    elif heights[q0] > heights[q1]:
        if bosses[q0] == q1:
            return q1
        else:
            return run(bosses[q0], q1, bosses, heights)
    else:
        if q0 == bosses[q1]:
            return q0
        else:
            return run(q0, bosses[q1], bosses, heights)

empCount, qCount = [int(aa) for aa in input().strip().split(" ")]
bosses = [0]+[int(aa)-1 for aa in input().strip().split(" ")]
heights = [-1]*len(bosses)
queryes = [tuple([int(aa)-1 for aa in input().strip().split(" ")]) for _ in range(qCount)]

for ii in range(len(heights)):
    kk = ii
    hh = 0
    while kk != 0:
        # print(kk,bosses[kk])
        kk = bosses[kk]
        hh += 1
    heights[ii] = hh
# comp = [Emp(ii, 0 if ii == 0 else bosses[ii-1]) for ii in range(empCount)]
# print(comp)
# print(bosses)
# print(heights)
# print(queryes)
for q0,q1 in queryes:
    print(run(q0, q1, bosses, heights)+1)
# bosses 

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