力扣解法汇总1019. 链表中的下一个更大节点

目录链接:

力扣编程题-解法汇总_分享+记录-CSDN博客

GitHub同步刷题项目:

https://github.com/September26/java-algorithms

原题链接:力扣


描述:

给定一个长度为 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

解题思路:

* 解题思路:
* 构建一个栈结构stack,从后向前遍历。
* 如果当前值>=栈顶的值,则从栈顶pop出这个值。继续取栈顶的值进行比较。
* 如果栈为空,则说明后面没有更大的值了,则设置当前位置的值位0。
* 如果当前值<栈顶的值,则从说明当前位置的更大值就是栈顶的值。

代码:

public class Solution1019 {

    public int[] nextLargerNodes(ListNode head) {
        Stack<Integer> stack = new Stack<>();
        List<Integer> list = new ArrayList<>();
        while (head != null) {
            list.add(head.val);
            head = head.next;
        }
        int[] arr = new int[list.size()];
        for (int i = list.size() - 1; i >= 0; i--) {
            Integer value = list.get(i);
            while (stack.size() > 0) {
                Integer peek = stack.peek();
                if (value >= peek) {
                    stack.pop();
                } else {
                    break;
                }
            }
            if (stack.size() == 0) {
                arr[i] = 0;
            } else {
                arr[i] = stack.peek();
            }
            stack.add(value);
        }
        return arr;
    }
}

猜你喜欢

转载自blog.csdn.net/AA5279AA/article/details/130054378