假设我们有一个价格列表,按时间顺序代表一家公司的每日股票市场价格。我们必须找到一个相同长度的列表,其中索引 i 处的值将是我们必须等到盈利的最少天数。如果没有这样的盈利方式,则该值应为 0。
因此,如果输入类似于价格 = [4, 3, 5, 9, 7, 6],那么输出将是 [2, 1, 1, 0, 0, 0]
为了解决这个问题,我们将按照以下步骤操作:
ans := 大小与价格相同并用 0 填充的列表
q := 一个新列表
对于价格中的每个指数 i 和价格 p,做
j := q 的最后一个元素的第一项
ans[j] := i - j
从 q 中删除最后一个元素
当 q 不为空且 p > q 的最后一项的第二个值时,做
在 q 的末尾插入 (i, p)
返回答案
让我们看下面的实现来更好地理解:
class Solution: def solve(self, prices): ans = [0 for _ in prices] q = [] for i, p in enumerate(prices): while q and p > q[-1][1]: j = q[-1][0] ans[j] = i - j q.pop() q.append((i, p)) return ans ob = Solution() prices = [4, 3, 5, 9, 7, 6] print(ob.solve(prices))
[4, 3, 5, 9, 7, 6]输出结果
[2, 1, 1, 0, 0, 0]