找出数组中每个数右边第一个比它大的元素--时间复杂度o(n)python单调栈解法

题目：给定一个整型数组，数组元素随机无序的，要求打印出所有元素右边第一个大于该元素的值。

如数组A=[1,5,3,6,4,8,9,10] 输出[5, 6, 6, 8, 8, 9, 10, -1]

如数组A=[8, 2, 5, 4, 3, 9, 7, 2, 5] 输出[9, 5, 9, 9, 9, -1, -1, 5, -1]

def find_max_right(arr):
    if not arr:
        return arr
    stack = []
    stack.append(0)
    result = [0 for i in range(len(arr))]
    for i in range(1,len(arr)):
        while stack:
            if arr[i] > arr[stack[-1]]:
                pos = stack.pop()
                result[pos] = arr[i]
            else:
                stack.append(i)
                break
        if not stack:
            stack.append(i)
    while stack:
        pos = stack.pop()
        result[pos] = -1

    return result

1、初始化栈，里面第一个元素为元素索引值0

2、遍历到下一个元素A[i],如果栈不为空且当前遍历的元素值A[i]大于栈顶的元素值，则说明当前元素正好是栈顶元素右边第一个比他大的元素，将栈顶元素弹出，result[stack.pop()]=A[i]，继续遍历元素值A[i]是否大于新的栈顶元素值，如果大于，说明A[i]也是这个元素右边第一个比他大的元素，将栈顶元素弹出&