update

Author: lazywhite

004.单调栈.go

@@ -0,0 +1,55 @@
+//
+// Copyright (C) 2022 lazywhite <[email protected]>
+//
+// Distributed under terms of the MIT license.
+//
+
+package main
+
+import (
+	"fmt"
+)
+
+func main() {
+	nums := []int{10, 3, 7, 4, 12}
+	ans := monoStack(nums)
+	fmt.Println(nums)
+	fmt.Println(ans)
+}
+
+/*
+1. 下一个, 从前向后遍历
+   上一个, 从后向前遍历
+2. smarller 弹出比v大的
+2. larger 弹出比v小的
+
+*/
+
+func monoStack(nums []int) []int {
+	// stack存储num的index
+	stack := make([]int, 0)
+	ans := make([]int, len(nums))
+
+	for i, v := range nums {
+		if len(stack) == 0 {
+			stack = append(stack, i)
+		} else {
+			for len(stack) > 0 && nums[stack[len(stack)-1]] > v {
+				// 更新结果
+				peek := stack[len(stack)-1]
+				ans[peek] = v
+				stack = stack[:len(stack)-1]
+			}
+			// 入栈
+			stack = append(stack, i)
+		}
+	}
+
+	// stack不为空
+	if len(stack) > 0 {
+		for _, idx := range stack {
+			ans[idx] = -1
+		}
+	}
+	return ans
+}

1475.商品折扣后的最终价格.go

@@ -0,0 +1,84 @@
+/*
+ * @lc app=leetcode.cn id=1475 lang=golang
+ *
+ * [1475] 商品折扣后的最终价格
+ *
+ * https://leetcode-cn.com/problems/final-prices-with-a-special-discount-in-a-shop/description/
+ *
+ * algorithms
+ * Easy (71.50%)
+ * Likes:    61
+ * Dislikes: 0
+ * Total Accepted:    19.3K
+ * Total Submissions: 27K
+ * Testcase Example:  '[8,4,6,2,3]'
+ *
+ * 给你一个数组 prices ，其中 prices[i] 是商店里第 i 件商品的价格。
+ *
+ * 商店里正在进行促销活动，如果你要买第 i 件商品，那么你可以得到与 prices[j] 相等的折扣，其中 j 是满足 j > i 且 prices[j]
+ * <= prices[i] 的 最小下标 ，如果没有满足条件的 j ，你将没有任何折扣。
+ *
+ * 请你返回一个数组，数组中第 i 个元素是折扣后你购买商品 i 最终需要支付的价格。
+ *
+ *
+ *
+ * 示例 1：
+ *
+ * 输入：prices = [8,4,6,2,3]
+ * 输出：[4,2,4,2,3]
+ * 解释：
+ * 商品 0 的价格为 price[0]=8 ，你将得到 prices[1]=4 的折扣，所以最终价格为 8 - 4 = 4 。
+ * 商品 1 的价格为 price[1]=4 ，你将得到 prices[3]=2 的折扣，所以最终价格为 4 - 2 = 2 。
+ * 商品 2 的价格为 price[2]=6 ，你将得到 prices[3]=2 的折扣，所以最终价格为 6 - 2 = 4 。
+ * 商品 3 和 4 都没有折扣。
+ *
+ *
+ * 示例 2：
+ *
+ * 输入：prices = [1,2,3,4,5]
+ * 输出：[1,2,3,4,5]
+ * 解释：在这个例子中，所有商品都没有折扣。
+ *
+ *
+ * 示例 3：
+ *
+ * 输入：prices = [10,1,1,6]
+ * 输出：[9,0,1,6]
+ *
+ *
+ *
+ *
+ * 提示：
+ *
+ *
+ * 1 <= prices.length <= 500
+ * 1 <= prices[i] <= 10^3
+ *
+ *
+ */
+
+// @lc code=start
+func finalPrices(prices []int) []int {
+	stack := make([]int, 0)
+	ans := make([]int, len(prices))
+	for i, v := range prices {
+		if len(stack) == 0 {
+			stack = append(stack, i)
+		} else {
+			for len(stack) > 0 && prices[stack[len(stack)-1]] >= v {
+				peekIdx := stack[len(stack)-1]
+				ans[peekIdx] = prices[peekIdx] - v
+				stack = stack[:len(stack)-1]
+			}
+			stack = append(stack, i)
+		}
+	}
+
+	for _, idx := range stack {
+		ans[idx] = prices[idx]
+	}
+	return ans
+}
+
+// @lc code=end
+