在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]