C语言力扣第42题之接雨水。四种方法——暴力、动态规划、栈、双指针

给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。

示例 1：

输入：height = [0, 1, 0, 2, 1, 0, 1, 3, 2, 1, 2, 1]
输出：6
解释：上面是由数组[0, 1, 0, 2, 1, 0, 1, 3, 2, 1, 2, 1] 表示的高度图，在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）。

示例 2：

输入：height = [4, 2, 0, 3, 2, 5]
输出：9

提示：

n == height.length
1 <= n <= 2 * 104
0 <= height[i] <= 105

通过次数531, 609
提交次数863, 255

来源：力扣（LeetCode）
链接：https ://leetcode.cn/problems/trapping-rain-water
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

方法 1：暴力

直观想法

直接按问题描述进行。对于数组中的每个元素，我们找出下雨后水能达到的最高位置，等于两边最大高度的较小值减去当前高度的值。

算法

1. 初始化 ans=0
2. 从左向右扫描数组：
   初始化 max_left=0
   从当前元素向左扫描并更新：max_left=max⁡(max_left,height[j])
   从当前元素向右扫描并更新：max_right=max⁡(max_right,height[j])
   将min⁡(max_left,max_right)−height[i]累加到 ans

时间复杂度较大为O（n^2）

方法 2：动态编程

直观想法

在暴力方法中，我们仅仅为了找到最大值每次都要向左和向右扫描一次。但是我们可以提前存储这个值。因此，可以通过动态编程解决。

算法

1.     找到数组中从下标 i 到最左端最高的条形块高度 left_max
2.     找到数组中从下标 i 到最右端最高的条形块高度 right_max。
3.     扫描数组 height 并更新答案：
4.     累加 min⁡(max_left[i],max_right[i])−height[i]到 ans上

时间空间复杂度均为O（n）

方法 3：栈

直观想法

我们可以不用像方法 2 那样存储最大高度，而是用栈来跟踪可能储水的最长的条形块。使用栈就可以在一次遍历内完成计算。

我们在遍历数组时维护一个栈。如果当前的条形块小于或等于栈顶的条形块，我们将条形块的索引入栈，意思是当前的条形块被栈中的前一个条形块界定。如果我们发现一个条形块长于栈顶，我们可以确定栈顶的条形块被当前条形块和栈的前一个条形块界定，因此我们可以弹出栈顶元素并且累加答案到 ans

算法

1. 使用栈来存储条形块的索引下标。
2. 遍历数组：
   -当栈非空且 height[current]>height[st.top()]：
       意味着栈中元素可以被弹出。弹出栈顶元素 top；
       计算当前元素和栈顶元素的距离，准备进行填充操作；distance=current−st.top()−1
       找出界定高度
       bounded_height=min⁡(height[current],height[st.top()])−height[top]
       往答案中累加积水量ans+=distance×bounded_height
   -将当前索引下标入栈
   -将 current 移动到下个位置

时间空间复杂度和方法二一样。

方法 4：使用双指针

直观想法

和方法 2 相比，我们不从左和从右分开计算，我们想办法一次完成遍历。
只要 right_max[i]>left_max[i]，积水高度将由 left_max 决定，类似地 left_max[i]>right_max[i]
所以我们可以认为如果一端有更高的条形块（例如右端），积水的高度依赖于当前方向的高度（从左到右）。当我们发现另一侧（右侧）的条形块高度不是最高的，我们则开始从相反的方向遍历（从右到左）。
我们必须在遍历时维护 left_max 和 right_max ，但是我们现在可以使用两个指针交替进行，实现 1 次遍历即可完成。

这里直接放代码了

#define _CRT_SECURE_NO_WARNINGS
#include <stdio.h>
#include <string.h>


int trap(int* height, int heightSize) 
{
	int ans = 0;
	int left = 0;
	int right = heightSize - 1;
	int premax = 0;
	int max = 0;
	int i = 0;
	while (left < right)
	{
		if (height[left] > height[right] && height[right] > max)
		{
			premax = max;
			max = height[right];
			for (i = left + 1; i <= right - 1; i++)
			{
				if (height[i] <= premax)
					ans = ans - premax + max;
				if (height[i] > premax&&height[i] < max)
					ans = ans + max - height[i];
			}
			right--;
		}

		if (height[left] <= height[right] && height[left] > max)
		{
			premax = max;
			max = height[left];
			for (i = left + 1; i <= right - 1; i++)
			{
				if (height[i] <= premax)
					ans = ans - premax + max;
				if (height[i] > premax&&height[i] < max)
					ans = ans + max - height[i];
			}
			left++;
		}

		if (height[left] <= max)
		{
			left++;
		}
		if (height[right] <= max)
		{
			right--;
		}
	}
	return ans;
}

int main()
{
	int height[] = { 0, 1, 0, 2, 1, 0, 1, 3, 2, 1, 2, 1 };
	int heightSize = sizeof(height) / sizeof(int);
	int ans = trap(height, heightSize);
	printf("%d", ans);
}