链表中的下一个更大节点
题目描述
给定一个长度为 n 的链表 head
对于列表中的每个节点,查找下一个 更大节点 的值。也就是说,对于每个节点,找到它旁边的第一个节点的值,这个节点的值 严格大于 它的值。
返回一个整数数组 answer ,其中 answer[i] 是第 i 个节点( 从1开始 )的下一个更大的节点的值。如果第 i 个节点没有下一个更大的节点,设置 answer[i] = 0 。
示例
Java代码
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode() {}
* ListNode(int val) { this.val = val; }
* ListNode(int val, ListNode next) { this.val = val; this.next = next; }
* }
*/
class Solution {
public int[] nextLargerNodes(ListNode head) {
if(head == null) {
return null;
}
ArrayList<Integer> list = new ArrayList<>();
int size = 0;
while(head != null) {
list.add(head.val);
size++;
head = head.next;
}
Stack<Integer> stack = new Stack<>();
int[] ans = new int[size];
for(int i = list.size() - 1; i >= 0; i--) {
while(!stack.empty() && stack.peek() <= list.get(i)) {
stack.pop();
}
ans[i] = stack.empty() ? 0 : stack.peek();
stack.push(list.get(i));
}
return ans;
}
}
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/next-greater-node-in-linked-list
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。