在 JavaScript 中查找到下一个更大元素的距离
问题
我们需要编写一个JavaScript函数,它接受一个数字数组arr作为第一个也是唯一的参数。
我们的函数应该为输入构造一个新的数组,其中每个对应的元素是到比当前元素更大的下一个元素的距离,如果当前元素的右侧没有更大的元素,我们应该为对应的元素压入0res数组中的元素,最后我们应该返回这个数组。
例如,如果函数的输入是
输入
const arr = [12, 13, 14, 11, 16, 10, 12, 17, 19, 18];
输出
const output = [1, 1, 2, 1, 3, 1, 1, 1, 0, 0];
输出说明
因为下一个比12大的元素是13,距离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]