目录
一、盛水最多的容器
二、 接雨水
2.1 - 前后缀分解
2.2 - 相向双指针
一、盛水最多的容器
题目描述:
给定一个长度为 n
的整数数组 height
。有 n
条垂线,第 i
条线的两个端点是 (i, 0)
和 (i, height[i])
。
找出其中的两条线,使得它们与 x
轴共同构成的容器可以容纳最多的水。
返回容器可以储存的最大水量。
说明:你不能倾斜容器。
示例 1:
输入:[1,8,6,2,5,4,8,3,7]
输出:49
解释:图中垂直线代表输入数组 [1,8,6,2,5,4,8,3,7]。在此情况下,容器能够容纳水(表示为蓝色部分)的最大值为 49。
示例 2:
输入:height = [1,1]
输出:1
提示:
-
n == height.length
-
2 <= n <= 10^5
-
0 <= height[i] <= 10^4
代码实现:
int maxArea(int* height, int heightSize)
{
int max = 0;
int left = 0;
int right = heightSize - 1;
while(left < right)
{
int l = right - left;
int h = height[left] < height[right] ? height[left++] : height[right--];
int area =l * h;
max = area > max ? area : max;
}
return max;
}
分析:
二、 接雨水
题目描述:
给定 n
个非负整数表示每个宽度为 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 * 10^4
-
0 <= height[i] <= 10^5
2.1 - 前后缀分解
int trap(int* height, int heightSize)
{
int* pre_max = (int*)malloc(sizeof(int) * heightSize);
int* suf_max = (int*)malloc(sizeof(int) * heightSize);
pre_max[0] = height[0];
for (int i = 1; i < heightSize; i++)
{
if (height[i] > pre_max[i - 1])
pre_max[i] = height[i];
else
pre_max[i] = pre_max[i - 1];
}
suf_max[heightSize - 1] = height[heightSize - 1];
for (int i = heightSize - 2; i >= 0; i--)
{
if (height[i] > suf_max[i + 1])
suf_max[i] = height[i];
else
suf_max[i] = suf_max[i + 1];
}
int sum = 0;
for (int i = 0; i < heightSize; i++)
{
int h = (pre_max[i] < suf_max[i] ? pre_max[i] : suf_max[i]) - height[i];
sum += h * 1;
}
free(pre_max);
free(suf_max);
return sum;
}
分析:
如果能求出每个宽度为 1
的柱子上所接的雨水,那么将所有的水量相加就能得到结果。
pre_max[i]
表示 height[0] ~ height[i]
中最高的柱子;suf_max[i]
表示 height[i] ~ height[heightSize - 1]
中最高的柱子。
第 i
根的柱子上所接的雨水则为:min(pre_max[i], suf_max[i]) - height[i]
。
2.2 - 相向双指针
int trap(int* height, int heightSize)
{
int sum = 0;
int left = 0, right = heightSize - 1;
int pre_max = 0, suf_max = 0;
while (left <= right)
{
pre_max = height[left] > pre_max ? height[left] : pre_max;
suf_max = height[right] > suf_max ? height[right] : suf_max;
if (pre_max < suf_max)
{
sum += pre_max - height[left];
left++;
}
else
{
sum += suf_max - height[right];
right--;
}
}
return sum;
}
分析:
pre_max
表示 height[0] ~ height[left]
中最高的柱子;suf_max
表示 height[right] ~ height[heightSize - 1]
中最高的柱子。
-
如果
pre_max < suf_max
,第left
根柱子上所借雨水的高度就由pre_max
决定,因为suf_max[left]
一定大于pre_max
。 -
如果
suf_max < pre_max
,则第right
根柱子上所借雨水的高度就由suf_max
决定,因为pre_max[right]
一定大于suf_max
。
相较于前后缀分解,使用相向双指针求解,时间复杂度依然是 O(n),而空间复杂度则变成了 O(1)。