Ion Flux Relabeling

def process_converter(max_idx, converter):
    prev_node = -1
    current_node = max_idx
    # The subtree is how many elements we have in total for a node
    subtree = max_idx

    if current_node == converter:
        return prev_node

    prev_node = current_node

    while subtree > 1:
        # As we go down through the tree, each level reduces the number by half, which is the same as shifting 1 bit
        subtree = subtree >> 1

        left = current_node - subtree - 1
        right = current_node - 1

        if converter == left or converter == right:
            return prev_node

        if converter < left:
            current_node = left
        elif converter > left:
            current_node = right

        prev_node = current_node

    return -1

def solution(h, q):
    max_idx = 2**h-1
    return [process_converter(max_idx, converter) for converter in q]

Source: livecodestream.dev

Similar codes

2 votes145 views2 answers
Convert generator to list python
Python
37 votes123 views4 answers
How to create miniconda environment
Python
1 vote131 views1 answer
Dataframe select entries that are in a list
Python
0 vote151 views1 answer
Python find equal rows of two numpy arrays
Python
10 votes130 views3 answers
Python print traceback from exception
Python
78 votes150 views6 answers
How to wait in python
Python
0 vote190 views1 answer
Python ValueError: zero-size array to reduction operation maximum which has no identity
Python
56 votes128 views2 answers
Convert date string to date time string python
Python
1 vote144 views3 answers
Discord.py aliases
Python
0 vote134 views1 answer
Python float till 2 decimal places
Python