diff --git a/src/main/java/com/cool/hot100/linkedlist/ListNode.java b/src/main/java/com/cool/hot100/linkedlist/ListNode.java index 04647fb..94f8b60 100644 --- a/src/main/java/com/cool/hot100/linkedlist/ListNode.java +++ b/src/main/java/com/cool/hot100/linkedlist/ListNode.java @@ -2,8 +2,8 @@ package com.cool.hot100.linkedlist; public class ListNode { - int val; - ListNode next; + public int val; + public ListNode next; ListNode() { } diff --git a/src/main/java/com/cool/ling_cha_mount/stack/Num1019.java b/src/main/java/com/cool/ling_cha_mount/stack/Num1019.java new file mode 100644 index 0000000..15083c2 --- /dev/null +++ b/src/main/java/com/cool/ling_cha_mount/stack/Num1019.java @@ -0,0 +1,29 @@ +package com.cool.ling_cha_mount.stack; + +import com.cool.hot100.linkedlist.ListNode; + +import java.util.Deque; +import java.util.LinkedList; + +public class Num1019 { + public int[] nextLargerNodes(ListNode head) { + Deque stack = new LinkedList<>(); + int len = 0; + ListNode node = head; + while (node != null) { + len++; + node = node.next; + } + int[] res = new int[len]; + node = head; + int i = 0; + while (node != null) { + while (!stack.isEmpty() && stack.peek()[0] < node.val) { + res[stack.pop()[1]] = node.val; + } + stack.push(new int[]{node.val, i++}); + node = node.next; + } + return res; + } +}