Files
LeetCode-Go/leetcode/1019.Next-Greater-Node-In-Linked-List/1019. Next Greater Node In Linked List.go
2020-08-07 17:06:53 +08:00

39 lines
777 B
Go

package leetcode
import (
"github.com/halfrost/LeetCode-Go/structures"
)
// ListNode define
type ListNode = structures.ListNode
/**
* Definition for singly-linked list.
* type ListNode struct {
* Val int
* Next *ListNode
* }
*/
// 解法一 单调栈
func nextLargerNodes(head *ListNode) []int {
res, indexes, nums := make([]int, 0), make([]int, 0), make([]int, 0)
p := head
for p != nil {
nums = append(nums, p.Val)
p = p.Next
}
for i := 0; i < len(nums); i++ {
res = append(res, 0)
}
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
}