刷这题之前先看:
【算法】【OD算法】【单调栈】找朋友-CSDN博客
【算法】【单调栈】【leetcode】1475. 商品折扣后的最终价格-CSDN博客
【算法】【单调栈】【leetcode】901. 股票价格跨度-CSDN博客
【算法】【单调栈】每日温度-CSDN博客
题目地址:. - 力扣(LeetCode)
题目描述:
给定一个长度为
n
的链表head
对于列表中的每个节点,查找下一个 更大节点 的值。也就是说,对于每个节点,找到它旁边的第一个节点的值,这个节点的值 严格大于 它的值。
返回一个整数数组
answer
,其中answer[i]
是第i
个节点( 从1开始 )的下一个更大的节点的值。如果第i
个节点没有下一个更大的节点,设置answer[i] = 0
。
示例 1:
输入:head = [2,1,5] 输出:[5,5,0]
示例 2:
输入:head = [2,7,4,3,5] 输出:[7,0,5,5,0]
提示:
- 链表中节点数为
n
1 <= n <= 104
1 <= Node.val <= 109
代码实现:
/**
* 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) {
//栈中存放数组下标和值 0->值 1->坐标
Stack<Integer[]> st = new Stack();
int [] temp = new int [10*10*10*10];
Integer len=0;
while(null != head){
while(!st.isEmpty()&&st.peek()[0]<head.val){
temp[st.pop()[1]]=head.val;
}
st.push(new Integer[]{head.val,len++});
head=head.next;
}
int [] ans = new int[len];
for(int i=0; i<len;i++){
ans[i]=temp[i];
}
return ans;
}
}