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

最新推荐文章于 2023-11-25 21:13:31 发布

guxin_duyin

最新推荐文章于 2023-11-25 21:13:31 发布

阅读量259

点赞数

分类专栏： LeetCode 文章标签：链表 leetcode 数据结构

本文链接：https://blog.csdn.net/guxin_duyin/article/details/125493867

版权

LeetCode 专栏收录该内容

154 篇文章 1 订阅

订阅专栏

这篇博客介绍了如何利用单调栈解决链表中寻找每个节点下一个更大节点值的问题。通过将链表转化为数组，然后遍历数组并维护一个单调栈，当栈顶元素小于当前元素时，更新栈顶元素对应的下一个更大节点值。最后返回结果数组。这种方法高效地解决了题目所给的链表问题。

摘要由CSDN通过智能技术生成

题目

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

题解

思路

使用单调栈，先把链表转化为数组；然后再依次把数组中的项放入栈中；同时校验栈顶的元素是否比当前元素小，如果比当前元素小，则当前元素就是下一个更大的元素；更新为这个值即可。

代码实现

/**
 * @param {ListNode} head
 * @return {number[]}
 */
const nextLargerNodes = function (head) {
    const array = [];
    // 将链表转化为数组
    while (head) {
        array.push(head.val);
        head = head.next;
    }
    // 新建单调栈
    const stack = [];
    // 建立结果映射,默认填充为0
    const result = array.map(() => 0)
    for (let i = 0, len = array.length; i < len; i++) {
        // 如果栈顶元素比当前元素小,则当前元素就是下一个更大的元素
        while (stack.length && array[stack[stack.length - 1]] < array[i]) {
            // 更新此下标的结果为当前元素
            result[stack[stack.length - 1]] = array[i];
            // 出栈
            stack.pop();
        }
        // 把下标存入栈中
        stack.push(i);
    }
    return result;
};