给出一个以头节点 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 。
示例 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]
提示:
- 对于链表中的每个节点,
1 <= node.val <= 10^9
- 给定列表的长度在
[0, 10000]
范围内
题目标签:Stack / Linked List
题目链接:LeetCode / LeetCode中国
Language | Runtime | Memory |
---|---|---|
java | 30 ms | 42.6 MB |
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
class Solution {
class Node {
int index;
int value;
Node(int index, int value) {
this.index = index;
this.value = value;
}
}
public int[] nextLargerNodes(ListNode head) {
ListNode p = head;
int n = 0;
while (p != null) {
n++;
p = p.next;
}
int[] res = new int[n];
p = head;
Stack<Node> stk = new Stack<>();
int idx = 0;
while (p != null) {
while (!stk.empty() && p.val > stk.peek().value) {
res[stk.pop().index] = p.val;
}
stk.push(new Node(idx, p.val));
p = p.next;
idx++;
}
return res;
}
}