leetcode 1019. 链表中的下一个更大节点

1. 题目链接 https://leetcode-cn.com/problems/next-greater-node-in-linked-list/

2. 题目描述

   1. 给出一个以头节点 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}) 。

      注意：在下面的示例中，诸如 [2,1,5] 这样的输入（不是输出）是链表的序列化表示，其头节点的值为 2，第二个节点值为 1，第三个节点值为 5

   2. 示例 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]
      

3. 解题思路

   1. 维护一个递减栈（栈顶元素最小），遍历链表，当前元素大于栈顶元素时，不断出栈并将栈顶元素在链表中的位置置为当前元素。将当前元素加入栈，因为是链表所以要同时保存索引信息。

4. 代码

   1. python

      class Solution:
          def nextLargerNodes(self, head):
              res, s = [0] * 10000, []
              i = 0
              while head:
                  tmp, head = head.val, head.next
                  while s and s[-1][0] < tmp:
                      res[s[-1][1]] = tmp
                      s.pop()
                  s.append([tmp, i])
                  i += 1
              return res[:i]