京东算法笔试-及时止损

求大佬指导问题出在哪,是超时,还是思路有问题。
q = []
for i in range(int(qnum)):
q.append(input())
print(suns)
def jszx(n,m,nums,k):#股票数量、一天能卖的数量、损失情况、卖哪几只股票
nums.sort()
nums = nums[0:k]
ans = 0 # 亏损数量
count = 0 #买了集股
i,day= n,1 #总共多少股,第几天
while i > 0 and count < k and nums:
if (k-count) >= m :
for j in range(m):
i -= 1
v = nums.pop(-1)
ans += vday
count += m
else:
for j in range(k-count):
v = nums.pop(-1)
ans += v
day
day += 1
return ans
for i in range(int(qnum)):
print(jszx(int(gpnum),int(snum),suns,int(q[i])))


版权声明:本文为wyyhuli原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。