接雨水
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
算法分析
接雨水这道题需要利用木桶效应来计算每个位置的接雨量。如果我们能知道某个位置左右两侧的最大柱子高度(要比该位置的柱子要高!),取较小的那一个,(较大的那个没影响,因为木桶效应!)减去该位置的柱子高度,就得到了该位置的接雨量。
可以使用暴力求解时间复杂度为O(n^2),也可以使用双指针优化时间复杂度为O(n)。
移动指针有两个作用:
- 遍历每个位置,记录接水量
- 更新左右柱子高度最大值
当两个指针相遇时,遍历结束。
算法实现
class Solution {
public:
int trap(vector<int>& height) {
// 初始化结果为0
int ans = 0;
// 初始化左右指针,分别指向数组的首尾
int left = 0, right = height.size(