leetcode.1019 链表中的下一个更大节点 - 单调栈 + 链表

1019. 链表中的下一个更大节点

题目:

思路:

查找右边第一个比自己大的数,一眼单调栈,再看看,发现是单调递减栈

则我们对每个被弹出的数记录取代它的值,入栈记录该【值,下标】

这样被弹出时,可以迅速找到出栈元素对应的下标

单调栈:【蓝桥杯集训9】单调栈、单调队列(模拟栈、模拟队列)专题(3 / 3)_Roye_ack的博客-CSDN博客

class Solution {
    public int[] nextLargerNodes(ListNode head) {
        Deque<int[]> stk=new LinkedList<>();
        //单调递减栈
        ListNode t=head,tt=head;
        List<Integer> idx=new ArrayList<>();
        while(t!=null)
        {
            idx.add(t.val);
            t=t.next;
        }
        int n=idx.size();
        int[] res=new int[n];
        for(int i=0;i<n;i++)
        {
            while(!stk.isEmpty()&&stk.peek()[0]<idx.get(i)) 
            {
                res[stk.peek()[1]]=idx.get(i);
                stk.pop();
            }
            stk.push(new int[]{idx.get(i),i});
        }
        return res;
    }
}

猜你喜欢

转载自blog.csdn.net/weixin_61639349/article/details/130063375
今日推荐