分奖金(python)

题目描述

老板要给员工分配奖金,分配方式通过游戏决定。安装员工工号排序,每个人随机抽取一个数字。按照工号的顺序往后排列,遇到第一个比自己大的数字,前面员工可以获得“距离*数字差值”的奖金,如果遇到不比自己打的,就给自己分配随机数量的奖金。

输入描述

第一行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

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
扎眼的阳光的头像扎眼的阳光普通用户
上一篇 2024年2月19日
下一篇 2024年2月19日

相关推荐