題目
給出一個以頭節點 head 作為第一個節點的連結串列。連結串列中的節點分別編號為:node_1, node_2, node_3, ... 。
每個節點都可能有下一個更大值(next larger value):對於 node_i,如果其 next_larger(node_i) 是 node_j.val,那麼就有 j > i 且 node_j.val > node_i.val,
而 j 是可能的選項中最小的那個。如果不存在這樣的 j,那麼下一個更大值為 0 。
返回整數答案陣列 answer,其中 answer[i] = next_larger(node_{i+1}) 。
注意:在下面的示例中,諸如 [2,1,5] 這樣的輸入(不是輸出)是連結串列的序列化表示,其頭節點的值為 2,
第二個節點值為 1,第三個節點值為 5 。
示例 1:輸入:[2,1,5] 輸出:[5,5,0]
示例 2:輸入:[2,7,4,3,5] 輸出:[7,0,5,5,0]
示例 3:輸入:[1,7,5,1,9,2,5,1] 輸出:[7,9,9,9,0,5,0,0]
提示:對於連結串列中的每個節點,1 <= node.val <= 10^9
給定列表的長度在 [0, 10000] 範圍內
解題思路分析1、暴力法;時間複雜度O(n^2),空間複雜度O(n)
func nextLargerNodes(head *ListNode) []int { arr := make([]int, 0) if head == nil { return arr } for head != nil { arr = append(arr, head.Val) head = head.Next } res := make([]int, len(arr)) for i := 0; i < len(arr); i++ { for j := i + 1; j < len(arr); j++ { if arr[i] < arr[j] { res[i] = arr[j] break } } } return res}
2、棧輔助;時間複雜度O(n),空間複雜度O(n)
func nextLargerNodes(head *ListNode) []int { arr := make([]int, 0) if head == nil { return arr } for head != nil { arr = append(arr, head.Val) head = head.Next } res := make([]int, len(arr)) stack := make([]int, 0) for i := 0; i < len(arr); i++ { for len(stack) > 0 && arr[i] > arr[stack[len(stack)-1]] { last := stack[len(stack)-1] res[last] = arr[i] stack = stack[:len(stack)-1] } stack = append(stack, i) } return res}
總結Medium題目,轉換為陣列後,然後使用暴力法或者單調棧計算即可
最新評論