Сегмент дерева мин и макс

Question

Я пытаюсь построить дерево сегментов, родительский узел которого должен содержать минимальное и максимальное значения его дочерних узлов. Теперь, когда я пытаюсь реализовать это, я сталкиваюсь с ошибкой, которая заключается в том, что один дочерний элемент может вернуть целое число, тогда как другой дочерний элемент может вернуть Список и работа функции max или min вызовет ошибку. Как это преодолеть?

from math import log2,ceil
def segment(low,high,pos):

    if (low==high):
        segment_tree[pos]=arr[low]
        return

    mid=(high+low)//2

    segment(low,mid,2*pos+1)
    segment(mid+1,high,2*pos+2)

    segment_tree[pos]=[min(segment_tree[2*pos+1],segment_tree[2*pos+2]),max(segment_tree[2*pos+1],segment_tree[2*pos+2])]

length=5
arr=[1,2,3,4,5]
low=0
high=length-1
height=int(ceil(log2(length)))
pos=0 
size_of_segment_tree=2*int(pow(2,height))-1
segment_tree=[0]*size_of_segment_tree
segment(low,high,pos)

Answer 1

Это будет работать с использованием isinstance(объект, список)

segment_tree[pos]=[min(min(segment_tree[2*pos+1]) if isinstance(segment_tree[2*pos+1],list) else segment_tree[2*pos+1],min(segment_tree[2*pos+2]) if isinstance(segment_tree[2*pos+2],list) else segment_tree[2*pos+2]),max(max(segment_tree[2*pos+1]) if isinstance(segment_tree[2*pos+1],list) else segment_tree[2*pos+1],max(segment_tree[2*pos+2]) if isinstance(segment_tree[2*pos+2],list) else segment_tree[2*pos+2])]

Изменить: переформатировать для ясности:

segment_tree[pos]=  [
    min(min(segment_tree[2*pos+1]) 
        if isinstance(segment_tree[2*pos+1],list) 
        else segment_tree[2*pos+1],
    min(segment_tree[2*pos+2]) 
        if isinstance(segment_tree[2*pos+2],list) 
        else segment_tree[2*pos+2]),
    max(max(segment_tree[2*pos+1]) 
        if isinstance(segment_tree[2*pos+1],list) 
        else segment_tree[2*pos+1],
    max(segment_tree[2*pos+2]) 
        if isinstance(segment_tree[2*pos+2],list) 
        else segment_tree[2*pos+2])
        ]

Answer 2

Вы могли бы написать функцию min/max самостоятельно, если я правильно понял проблему:

def my_min(value):
    if isinstance(value, list):
        return min(value)
    else:
        return value

def my_max(value):
    if isinstance(value, list):
        return max(list)
    else:
        return value