计算数组中每个数左边/右边第一个比其大/小的值 python

计算数组中每个数左边第一个比其大的值
如果用最简单的暴力法，时间复杂度最坏情况下 O(n^2)，遍历每一个数，再遍历它的两边

arr = [5,2,4,9,3,1,5,7,8,6,1,3,5,5]
stack = []
n = len(arr)
leftFirstMax = [None]*n
rightFirstMax = [None]*n
for i in range(n):
	p = i-1 
	q = i+1
	for p in range(i-1,-1,-1):
		if arr[p] > arr[i]:
			break
	if p>0 or (p==0 and arr[p]>arr[i]):
		leftFirstMax[i] = arr[p]
	for q in range(i+1,n):
		if arr[q] > arr[i]:
			break
	if q<n-1 or (q==n-1 and arr[q]>arr[i]):
		rightFirstMax[i] = arr[q]
print(leftFirstMax)
print(rightFirstMax)
	

用栈解决，遍历到a[i]
当栈中为空，直接压入
栈不为空，比较栈顶元素 top 和 a[i]。 若 top < a[i] ，弹出栈顶元素。循环执行，直到遇到第一个 top>a[i] （top即为第一个左边比其大的元素）或者 栈为空（左边没有比 a[i] 大的元素）
栈中的元素，另其弹出的数，若大于其，则为该栈中元素右边最近的大的值
因此栈中元素从底到上按照从大到小的顺序，单调栈

arr = [5,2,4,9,3,3,5,7,8,6,8,5,5,5,7]
stack = []
n = len(arr)
leftFirstMax = [None]*n
rightFirstMax = [None]*n
for i in range(n):
    samei = []
    while len(stack)!=0 and arr[i]>=arr[stack[-1]]:
        m = stack.pop()
        if arr[i] > arr[m]:
            rightFirstMax[m] = arr[i] #第m个元素要被弹出，且第i个大于第m个，第i个数为m右边最大的
        else:
            samei.append(m) #相同的先存起来，后面在push回去
    if len(stack)!=0:
        leftFirstMax[i] = arr[stack[-1]]  #stack中最后的为其左边最大的
    while len(samei)!=0:
        stack.append(samei.pop())
    stack.append(i)
print(leftFirstMax)
print(rightFirstMax)