在python中查找等待获利的最少天数的程序

假设我们有一个价格列表,按时间顺序代表一家公司的每日股票市场价格。我们必须找到一个相同长度的列表,其中索引 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]