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

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

1. 问题描述

2. 思路

2.1. 思路一

1. 存到数组中

2.2. 思路二

1. 反转链表
2. 单调栈

3. 代码

/**
 * Definition for singly-linked list.
 * type ListNode struct {
 *     Val int
 *     Next *ListNode
 * }
 */
func nextLargerNodes(head *ListNode) []int {
    var res []int
    p := head

    // 1. 存到数组里
    var nums []int
    for p != nil {
        nums = append(nums, p.Val)
        p = p.Next
    }

    for i := 0; i < len(nums); i++ {
        res = append(res, 0)
    }

    var indexes []int
    for i := 0; i < len(nums); i++ {
        
        num := nums[i]
        for len(indexes) > 0 && nums[indexes[len(indexes) - 1]] < num {
            index := indexes[len(indexes) - 1]
            res[index] = num
            indexes = indexes[:len(indexes) - 1]
        }
        indexes = append(indexes, i)
    }
    return res
}