给出一个以头节点 head 作为第一个节点的链表。链表中的节点分别编号为:node_1, node_2, node_3, … 。
每个节点都可能有下一个更大值(next larger value):对于 node_i,如果其 next_larger(node_i) 是 node_j.val,那么就有 j > i 且 node_j.val > node_i.val,而 j 是可能的选项中最小的那个。如果不存在这样的 j,那么下一个更大值为 0 。
返回整数答案数组 answer,其中 answer[i] = next_larger(node_{i+1}) 。
示例 1:
输入:[2,1,5]
输出:[5,5,0]
示例 2:
输入:[2,7,4,3,5]
输出:[7,0,5,5,0]
示例 3:
输入[1,7,5,1,9,2,5,1]
输出[7,9,9,9,0,5,0,0]
解析
将链表转化为一个数组,求数组中每个元素之后第一个最大值。暴力解法,需O(n2)。但其中会有大量的重复比较,可能过单调栈来解决。
考虑如下情况:
[4,1,9,2,5,1]
自后向前遍历上述数组,栈顶元素即保存了当前元素之后单调递增的一个序列。如果第i个元素比栈顶元素大,则栈顶元素即为该元素之后第一个比其大的值;如果比栈顶元素小或者相同,则在比较过程中,在计算1 - i个元素的过程中,栈中比当前元素小的可不用参与比较(即如果比第i个元素大了,肯定比这些大,如果比i小,则最多会计算到i。这也就是为什么可以采用单调栈),因此,将其出栈直到栈为空(表明无比第i个元素大的值)或栈顶元素比当前元素大。
重复操作,即可得。
代码
class Solution:
def nextLargerNodes(self, head: ListNode) -> List[int]:
from collections import deque
temp = head
out = []
while temp:
out.append(temp.val)
temp = temp.next
s = []
n = len(out)
for index in range(n - 1, -1, -1):
temp_v = out[index]
if not s:
s.append(temp_v)
out[index] = 0
else:
while s and temp_v >= s[-1]:
s.pop()
if not s:
out[index] = 0
else:
out[index] = s[-1]
s.append(temp_v)
return out