题目如下:
这道题目可以使用双指针和动态规划两种方法来解决。下面分别介绍这两种解题思路:
- 双指针
双指针法的思路是维护两个指针 left 和 right,分别从左右两端向中间移动,同时维护两个变量 left_max 和 right_max 表示左侧和右侧的最大高度,以及一个变量 ans 表示雨水的总量。遍历数组时,如果 height[left] < height[right],则表示左侧的柱子较低,此时可以确定左侧柱子可以接到的雨水量,即 left_max - height[left],然后将 left 指针向右移动一位,更新 left_max。右侧柱子同理。直到 left 和 right 指针重合时结束遍历,此时 ans 即为所求。
- 动态规划
动态规划法的思路是先遍历一遍数组,对于每个位置 i,分别求出左侧和右侧的最大高度 left_max 和 right_max,然后计算出该位置能接到的雨水量,即 min(left_max, right_max) - height[i],最后将每个位置的雨水量相加即可得到总雨水量。
以上两种方法时间复杂度均为 O(n),空间复杂度为 O(1)。
代码如下
双指针:
public int trap(int[] height) {
int n = height.length;
int left = 0, right = n - 1;
int leftMax = 0, rightMax = 0;
int ans = 0;
while (left < right) {
if (height[left] < height[right]) {
if (height[left] >= leftMax) {
leftMax = height[left];
} else {
ans += leftMax - height[left];
}
left++;
} else {
if (height[right] >= rightMax) {
rightMax = height[right];
} else {
ans += rightMax - height[right];
}
right--;
}
}
return ans;
}
其中,n 表示数组的长度,left 和 right 分别表示左右指针,leftMax 和 rightMax 分别表示左侧和右侧的最大高度,ans 表示雨水的总量。首先将 left 和 right 指向数组的两端,然后每次将较小的柱子向中间移动,并更新 leftMax 或 rightMax 的值,同时计算该位置能接到的雨水量并累加到 ans 中,直到 left 和 right 指针重合时结束遍历。最后返回 ans 即为所求。
动态规划:
public int trap(int[] height) {
int n = height.length;
if (n == 0) {
return 0;
}
int[] leftMax = new int[n];
int[] rightMax = new int[n];
leftMax[0] = height[0];
rightMax[n-1] = height[n-1];
for (int i = 1; i < n; i++) {
leftMax[i] = Math.max(leftMax[i-1], height[i]);
}
for (int i = n - 2; i >= 0; i--) {
rightMax[i] = Math.max(rightMax[i+1], height[i]);
}
int ans = 0;
for (int i = 0; i < n; i++) {
int level = Math.min(leftMax[i], rightMax[i]);
ans += level - height[i];
}
return ans;
}
其中,n 表示数组的长度,leftMax 和 rightMax 分别表示左侧和右侧的最大高度,ans 表示雨水的总量。首先判断数组是否为空,如果为空则直接返回 0。然后分别遍历数组两次,分别求出每个位置左侧和右侧的最大高度,分别保存在 leftMax 和 rightMax 数组中。最后再遍历一次数组,依次计算每个位置能接到的雨水量并累加到 ans 中。最后返回 ans 即为所求。
这道题目的关键点:
抓住这道题的关键点主要有两个:
- 对于某个位置而言,它能接到的雨水量取决于它左侧和右侧的最大高度中较小的那个,而不是最大的那个。因为如果取最大值,则可能会出现左右两侧高度不一致的情况,导致接不住雨水。
- 双指针法和动态规划法都是比较高效的解法,其中双指针法的时间复杂度为 O(n),空间复杂度为 O(1),而动态规划法的时间复杂度为 O(n),空间复杂度为 O(n)。
- 掌握了这两个关键点,就可以较好地理解并解决这道题目。
- 总的来说,这道题目既考察了基本的数据结构和算法知识,也考察了解决问题的思维能力和创新能力,是一道比较典型的算法面试题目。