【每日一题Day173】LC1019链表中的下一个更大节点 |单调栈

链表中的下一个更大节点【LC1019】

给定一个长度为 n 的链表 head

对于列表中的每个节点，查找下一个 更大节点 的值。也就是说，对于每个节点，找到它旁边的第一个节点的值，这个节点的值 严格大于 它的值。

返回一个整数数组 answer ，其中 answer[i] 是第 i 个节点( 从1开始 )的下一个更大的节点的值。如果第 i 个节点没有下一个更大的节点，设置 answer[i] = 0 。

好久没做单调栈啦

• 思路

  首先遍历一遍链表，将链表中的元素存储在集合中，然后使用单调递减栈找到严格大于在栈顶元素的下一个元素，当当前元素大于栈顶元素时，将栈顶元素弹出，并记录结果，因此栈中需要记录二元组{下标,值}

• 实现

  class Solution {
      public int[] nextLargerNodes(ListNode head) {
          ListNode cur = head;
          List<Integer> nums = new ArrayList<>();
          while (cur != null){
              nums.add(cur.val);
              cur = cur.next;
          }
          int n = nums.size();
          Deque<int[]> st = new LinkedList<>();
          int[] res = new int[n];
          for (int i = 0; i < n; i++){
              while (!st.isEmpty() && st.peekLast()[1] < nums.get(i)){
                  res[st.pollLast()[0]] = nums.get(i);
              }
              st.addLast(new int[]{i, nums.get(i)});
              
          }
          return res;
      }
  }
  

  • 复杂度
    • 时间复杂度：$O(n)$
    • 空间复杂度：$O(n)$