为 JavaScript 中的下一个更大元素查找距离


问题

我们需要编写一个 JavaScript 函数,其中接收一个数字数组 arr 作为第一个也是唯一的参数。

我们的函数应该为输入构造一个新数组,其中每个相应元素都是到下一个大于当前元素的元素的距离,如果没有大于当前元素的元素在它的右边,我们应该在 res 数组中为该相应元素添加 0,最后应该返回该数组。

例如,如果输入函数的是

输入

const arr = [12, 13, 14, 11, 16, 10, 12, 17, 19, 18];

输出

const output = [1, 1, 2, 1, 3, 1, 1, 1, 0, 0];

输出解释

因为 13 是大于 12 的下一个元素,距离为 1 块,

大于 13 的下一个元素是 14,距离为 1 块,

大于 14 的下一个元素是 16,距离为 2 块,依此类推。

以下是代码 −

示例

 实际演示

const arr = [12, 13, 14, 11, 16, 10, 12, 17, 19, 18];
const findNextGreater = (arr = []) => {
   const stack = []
   const res = new Array(arr.length).fill(0)
   for (let i = 0; i < arr.length; i++) {
      while (arr[i] > arr[stack[stack.length - 1]] && stack.length > 0) {
         const index = stack.pop()
         res[index] = i - index
      }
      stack.push(i)
   };
   return res
};
console.log(findNextGreater(arr));

输出

[1, 1, 2, 1, 3, 1, 1, 1, 0, 0]

更新日期:2021-4-24

191 次浏览

开启你的职业生涯

完成课程并获得认证

开始学习
广告