题目描述
老板要给员工分配奖金,分配方式通过游戏决定。安装员工工号排序,每个人随机抽取一个数字。按照工号的顺序往后排列,遇到第一个比自己大的数字,前面员工可以获得“距离*数字差值”的奖金,如果遇到不比自己打的,就给自己分配随机数量的奖金。
输入描述
第一行n表示员工数量(包含最后一个老板)
第二行每位员工分配的随机数字
示例
3
2 10 3
输出
8 10 3
解题思路
单调栈
参考https://leetcode.cn/problems/daily-temperatures/
import copy
def assignBonus():
staff = int(input())
nums = list(map(int, input().split()))
stack = []
ans = copy.deepcopy(nums)
for i in range(len(nums)):
while stack and nums[i] > nums[stack[-1]]:
preIndex = stack.pop()
ans[preIndex] = (i-preIndex)*(nums[i]-nums[preIndex])
stack.append(i)
print(ans)
版权声明:本文为博主作者:south_sheng原创文章,版权归属原作者,如果侵权,请联系我们删除!
原文链接:https://blog.csdn.net/south_sheng/article/details/128997848