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

题目

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

运行结果