代码随想录算法训练营第53天 | 977.有序数组的平方 、27. 移除元素
- 每日温度
今天正式开始单调栈,这是单调栈一篇扫盲题目,也是经典题。
大家可以读题,思考暴力的解法,然后在看单调栈的解法。 就能感受出单调栈的巧妙
https://programmercarl.com/0739.每日温度.html
单调栈确实很巧妙,在循环过程中记录左边或右边比他大或小的第一个值
/**
* @param {number[]} temperatures
* @return {number[]}
*/
var dailyTemperatures = function(temperatures) {
let len = temperatures.length;
const res = new Array(len).fill(0);
const stack = [];
stack[0] = 0;
for (let i=1;i<len;i++) {
let item = temperatures[i];
while(stack.length !== 0 && item > temperatures[stack[stack.length-1]]){
let index = stack.pop();
res[index] = i - index;
}
stack.push(i);
}
return res;
};
496.下一个更大元素 I
本题和 739. 每日温度 看似差不多,其实 有加了点难度。
https://programmercarl.com/0496.下一个更大元素I.html
和上题差不多,变化在于用一个map记录各个数字的对于的坐标
/**
* @param {number[]} nums1
* @param {number[]} nums2
* @return {number[]}
*/
var nextGreaterElement = function(nums1, nums2) {
const map = new Map();
for (let i=0;i<nums1.length;i++) {
map.set(nums1[i], i);
}
const res = new Array(nums1.length).fill(-1);
const stack = [0];
for (let i=1;i<nums2.length;i++) {
while(stack.length && nums2[i] > nums2[stack[stack.length-1]]) {
let index = stack.pop();
let index1 = map.get(nums2[index]);
if (index1>=0 && index1!==null) {
res[index1] = nums2[i];
}
}
stack.push(i);
}
return res;
};
503.下一个更大元素II
这道题和 739. 每日温度 几乎如出一辙,可以自己尝试做一做
https://programmercarl.com/0503.下一个更大元素II.html
可以用两个数组拼接在一起当作循环,也可以用i%length模拟循环
/**
* @param {number[]} nums
* @return {number[]}
*/
var nextGreaterElements = function(nums) {
const len = nums.length;
const res = new Array(len).fill(-1);
const stack = [0];
for (let i=1;i<len*2;i++) {
while(stack.length && nums[stack[stack.length-1]] < nums[i%len]){
let index = stack.pop();
res[index] = nums[i%len];
}
stack.push(i%len);
}
return res;
};