找出数组中每个数右边第一个比它大的元素--时间复杂度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]也是这个元素右边第一个比他大的元素，将栈顶元素弹出，result[stack.pop()]=A[i]，一直循环，直到栈顶为空或者当前遍历的元素值小于栈顶元素索引处的值

如果栈为空，则说明前面的元素都找到了比他大的元素，直接将当前元素的索引放入栈中

如果当前遍历的元素值A[i]小于栈顶元素索引的值，则说明还未找到栈顶元素中右边第一个比他大的元素，直接将当前遍历元素的索引入栈即可

3、直到遍历完所有元素，如果栈不为空，则说明栈中保存的全是未找到右边第一个比他大的数组索引，我们依次将这些元素出栈，并赋值result[stack.pop()]=-1