程式碼隨想錄演算法訓練營第55天 | 42. 接雨水 、84.柱狀圖中最大的矩形

YuanYF6發表於2024-07-07
  1. 接雨水

接雨水這道題目是 面試中特別高頻的一道題,也是單調棧 應用的題目,大家好好做做。
建議是掌握 雙指標 和單調棧,因為在面試中 寫出單調棧可能 有點難度,但雙指標思路更直接一些。
在時間緊張的情況有,能寫出雙指標法也是不錯的,然後可以和麵試官在慢慢討論如何最佳化。
https://programmercarl.com/0042.接雨水.html

/**
 * @param {number[]} height
 * @return {number}
 */
var trap = function(height) {
    if (height.length <= 2) {
        return 0;
    }
    const len = height.length;
    const stack = [0];
    let res = 0;
    for (let i=1; i<len;i++) {
        while(stack.length && height[stack[stack.length - 1]] < height[i]) {
            let mid = stack.pop();
            if (stack.length >= 1) {
                let left = stack[stack.length - 1];
                let min = Math.min(height[i], height[left]);
                res += (i- left -1) * (min - height[mid]);
            }
        }
        stack.push(i);
    }
    return res;
};

84.柱狀圖中最大的矩形
有了之前單調棧的鋪墊,這道題目就不難了。
https://programmercarl.com/0084.柱狀圖中最大的矩形.html

/**
 * @param {number[]} heights
 * @return {number}
 */
var largestRectangleArea = function(heights) {
    let maxArea = 0;
	const stack = [0];
	heights.push(0);
	const n = heights.length;

	for (let i = 1; i < n; i++) {
        let top = stack.at(-1)
        if (heights[top] < heights[i]) {
            stack.push(i);
        } else if (heights[top] === heights[i]) {
            stack.push(i);
        } else {
            while(stack.length && heights[top] > heights[i]) {
                const h = heights[stack.pop()];
                const left = stack.at(-1)?? -1;
                const w = i - left -1;
                maxArea = Math.max(maxArea, w*h);
                top = stack.at(-1);
            }
            stack.push(i)
        }
    }
	return maxArea;
};

相關文章