华为OD机试 Python【转盘寿司】

题目

寿司店周年庆，正在举办优惠活动回馈新老客户。
寿司转盘上总共有 n 盘寿司，prices[i] 是第 i 盘寿司的价格，
如果客户选择了第 i 盘寿司，寿司店免费赠送客户距离第 i 盘寿司最近的下一盘寿司 j，前提是 prices[j] < prices[i]，如果没有满足条件的 j，则不赠送寿司。
每个价格的寿司都可无限供应。

输入描述
输入的每一个数字代表每盘寿司的价格，每盘寿司的价格之间使用空格分隔，例如:
3 15 6 14

表示：
第 0 盘寿司价格 prices[0] 为 3
第 1 盘寿司价格 prices[1] 为 15
第 2 盘寿司价格 prices[2] 为 6
第 3 盘寿司价格 prices[3] 为 14
寿司的盘数 n 范围为：1 ≤ n ≤ 500
每盘寿司的价格 price 范围为：1 ≤ price ≤ 1000

输出描述
输出享受优惠后的一组数据，每个值表示客户选择第 i 盘寿司时实际得到的寿司的总价格。使用空格进行分隔，例如：
3 21 9 17

解题思路

一个栈来跟踪每个元素的索引。栈中的元素按照价格单调递减排列，即从栈底到栈顶的元素价格依次递减。对输入数组进行两次循环。第一次循环用于遍历数组中的每个元素，第二次循环用于确保每个元素都有机会与其后的元素进行比较。在遍历过程中，如果当前元素比栈顶元素小，则将栈顶元素出栈，并且更新结果数组中对应位置的值。这是因为找到了栈顶元素的下一个更小值，所以可以将其加到结果中。

代码