原题连接:leetcode-cn.com/problems/tr…
解题思路:
- 以该题的示例1为例,使用栈的解法,就是直观的去求所有柱子可装水的量。
- 栈中存储的一直是水桶的左边界和桶底的索引,例如分别会存储索引
[1, 2]
、[3, 4, 5]
、[7, 8, 9]
。 - 当遍历到柱子比栈顶高时,就将栈顶元素作为桶底出栈,出栈后剩下的栈顶即为左边界,以此形成水桶的一层,即可计算装水量,如下图:
- 需要注意如果桶底元素出栈之后,栈为空,表示无法组成一个桶,例如桶底元素的索引是0,需要退出出栈循环。
/**
* @param {number[]} height
* @return {number}
*/
var trap = function (height) {
let result = 0; // 储存总转水量
let stack = []; // 栈中存储的是水桶的左边界和桶底
// 遍历所有的柱子,在入栈之前,柱子都是作为水桶的右边界
for (let right = 0; right < height.length; right++) {
// 如果右边界比栈顶元素高,表示其与栈中元素形成了一个水桶,可以计算水桶的装水量
while (height[right] > height[stack[stack.length - 1]]) {
// 栈顶元素即为当前水桶的桶底,将其弹出并计算桶底高度
const bucketBottom = height[stack.pop()];
// 将桶底出栈之后,栈如果为空则表示不存在水桶,退出循环
if (!stack.length) {
break;
}
const left = stack[stack.length - 1]; // 剩余的栈顶元素为水桶的左边界,获取其索引
const leftHeight = height[left]; // 获取水桶左边界的高度
const rightHeight = height[right]; // 获取水桶右边界的高度
const bucketTop = Math.min(leftHeight, rightHeight); // 水桶左右边界的较小值,即为桶顶高度
const bucketHeight = bucketTop - bucketBottom; // 水桶的可装水高度为桶顶减去桶底高度
const width = right - left - 1; // 水桶的实际宽度为左右边界之差减一
const area = width * bucketHeight; // 计算当前水桶可装水量
result += area; // 将水量加入总量
}
// 当前右边界可以作为下一个水桶的左边界,因此需要入栈
stack.push(right);
}
return result;
};
常见问题FAQ
- 免费下载或者VIP会员专享资源能否直接商用?
- 本站所有资源版权均属于原作者所有,这里所提供资源均只能用于参考学习用,请勿直接商用。若由于商用引起版权纠纷,一切责任均由使用者承担。更多说明请参考 VIP介绍。
- 提示下载完但解压或打开不了?
- 找不到素材资源介绍文章里的示例图片?
- 模板不会安装或需要功能定制以及二次开发?
发表评论
还没有评论,快来抢沙发吧!