【OD机试题笔记】转盘寿司

题目描述

寿司店周年庆，正在举办优惠活动回馈新老用户。

寿司转盘上总共有 n 盘寿司， prices[i] 是第 i 盘寿司的价格。

如果客户选择了第 i 盘寿司， 寿司店免费赠送客户距离第 i 盘寿司最近的下一盘寿司 j ，前提是 prices[j] < prices[i]，如果没有满足条件的 i ，则不赠送寿司。

每个价格的寿司都可无限供应。

输入描述
输入的每一个数字代表寿司的价格，每盘寿司的价格之间使用空格分隔，例如：
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

示例1
输入：
3 15 6 14

输出：
3 21 9 17

思路

1. 核心逻辑：用单调栈找循环数组中每个元素的下一个更小元素，计算当前元素与更小元素的和（无则仅保留当前值）；
2. 实现步骤：
   • 遍历2n个元素（i从0到2n-1），通过i%n映射原数组索引，模拟环形遍历；
   • 维护单调递增栈（存索引），若当前价格<栈顶索引对应价格，更新栈顶元素的总价格（+当前价格）并弹出栈顶；
   • 仅前n个索引入栈，避免重复处理；
3. 时空复杂度：时间O(n)（每个元素仅入栈/出栈一次），空间O(n)（栈和结果数组均占用n空间）。

代码

function solution() {
    const prices = readline().split(' ').map(Number);
    const n = prices.length;
    const ans = [...prices]; 
    const stack = [];

    // 遍历2n个元素，模拟环形
    for (let i = 0; i < 2 * n; i++) {
        const idx = i % n;
        // 保持栈单调递增，找到最近更小元素
        while (stack.length && prices[stack[stack.length - 1]] > prices[idx]) {
            const topIdx = stack.pop();
            ans[topIdx] += prices[idx];
        }
        // 只push前n个索引，避免重复处理
        if (i < n) stack.push(idx);
    }

    console.log(ans.join(' '));
}

cases = [
    `3 15 6 14`,
    `4 5 6 3`,
    `3 7 5 6`,
    `5 3 6 4`,
    `6 8 7 9 5`
];

let caseIndex = 0;
let lineIndex = 0;

const readline = (function () {
  let lines = [];
  return function () {
    if (lineIndex === 0) {
      lines = cases[caseIndex]
        .trim()
        .split("\n")
        .map((line) => line.trim());
    }
    return lines[lineIndex++];
  };
})();

cases.forEach((_, i) => {
  caseIndex = i;
  lineIndex = 0;
  solution();
  console.log('-------');
});