程式碼隨想錄演算法訓練營第五十九天 | 42. 接雨水，503下一個更大元素

勉g發表於2024-03-29
演算法

503. 下一個更大元素 II

已解答
中等

相關標籤

相關企業

給定一個迴圈陣列 nums （ nums[nums.length - 1] 的下一個元素是 nums[0] ），返回 nums 中每個元素的 下一個更大元素 。

數字 x 的 下一個更大的元素 是按陣列遍歷順序，這個數字之後的第一個比它更大的數，這意味著你應該迴圈地搜尋它的下一個更大的數。如果不存在，則輸出 -1 。

示例 1:

輸入: nums = [1,2,1]
輸出: [2,-1,2]
解釋: 第一個 1 的下一個更大的數是 2；
數字 2 找不到下一個更大的數； 
第二個 1 的下一個最大的數需要迴圈搜尋，結果也是 2。

示例 2:

輸入: nums = [1,2,3,4,3]
輸出: [2,3,4,-1,4]

提示:

  • 1 <= nums.length <= 104
  • -109 <= nums[i] <= 109

package main

func nextGreaterElements(nums []int) []int {
nums2 := append(nums, nums...)
var momoStack []int
res := make([]int, len(nums2))
for i := 0; i < len(nums); i++ {
res[i] = -1
}
for i := 0; i < len(nums2); i++ {
for len(momoStack) != 0 && nums2[momoStack[len(momoStack)-1]] < nums2[i] {
res[momoStack[len(momoStack)-1]] = nums2[i]
momoStack = momoStack[:len(momoStack)-1]
}
momoStack = append(momoStack, i)
}
return res[:len(nums)]
}

42. 接雨水

已解答
困難

相關標籤

相關企業

給定 n 個非負整數表示每個寬度為 1 的柱子的高度圖，計算按此排列的柱子，下雨之後能接多少雨水。

示例 1：

程式碼隨想錄演算法訓練營第五十九天 | 42. 接雨水，503下一個更大元素
輸入：height = [0,1,0,2,1,0,1,3,2,1,2,1]
輸出：6
解釋：上面是由陣列 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度圖，在這種情況下，可以接 6 個單位的雨水（藍色部分表示雨水）。

示例 2：

輸入：height = [4,2,0,3,2,5]
輸出：9

提示：

  • n == height.length
  • 1 <= n <= 2 * 104
  • 0 <= height[i] <= 105

package main
func trap(height []int) int {
momoStack := make([]int, len(height))
var res int
for i := 0; i < len(height); i++ {
for len(momoStack) != 0 && height[momoStack[len(momoStack)-1]] < height[i] {
cur := momoStack[len(momoStack)-1]
momoStack = momoStack[:len(momoStack)-1]
if len(momoStack) != 0 {
res += (i - momoStack[len(momoStack)-1] - 1) *
(min(height[momoStack[len(momoStack)-1]], height[i])- height[cur])
}
}
momoStack = append(momoStack, i)
}
return res
}
func min(a, b int) int {
if a > b {
return b
}
return a
}

相關文章