503. 下一个更大元素 II
给定一个循环数组 nums
( nums[nums.length - 1]
的下一个元素是 nums[0]
),返回 nums
中每个元素的 下一个更大元素 。
数字 x
的 下一个更大的元素 是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1
。
示例 1:
输入: nums = [1,2,1] 输出: [2,-1,2] 解释: 第一个 1 的下一个更大的数是 2; 数字 2 找不到下一个更大的数; 第二个 1 的下一个最大的数需要循环搜索,结果也是 2。
示例 2:
输入: nums = [1,2,3,4,3] 输出: [2,3,4,-1,4]
提示:
1 <= nums.length <= 104
-109 <= nums[i] <= 109
和739. 每日温度,496.下一个更大元素 I 里下一个更大元素 I 的思路基本一样,把数组复制一份拼接到原数组后,达到类似于循环的效果。或者模拟遍历两次数组,可以节省空间和时间。
class Solution {
public int[] nextGreaterElements(int[] nums) {
int len = nums.length;
int[] res = new int[len];
Arrays.fill(res,-1);
Deque<Integer> stack = new LinkedList<>();
for (int i = 0; i < 2 * len; i++) {
while (!stack.isEmpty() && nums[i % len] > nums[stack.peek()]) {
res[stack.peek()] = nums[i % len];
stack.pop();
}
stack.push(i % len);
}
return res;
}
}
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
非常经典的一道题目,基本是刷题必刷。可以先试试用双指针完成这道题目。完成这道题目核心在于理解“一只水桶能盛多少水,取决于最短的那块木板” 。本题就可以看成是多个二维的水桶的拼接,接水的高度取决于两侧的“木板”最“短”的那个。需要注意的是取的是两侧的木板最短的,两侧木板中间的高低起伏,只要没高过两侧木板,是不影响水的最大高度的。而如果中间有一块比两侧木板还高的,那我们就不将其看成一个木桶,而是两个木桶的拼接,两个木桶较高的一侧为中间那个更高的柱子。所以思路应该是遍历数组,然后在遍历过程中向右向左遍历,找出当前元素右侧和左侧最高的柱子作为“木桶”的两侧“木板”,取最短的那个作为木桶装水的最大高度,再减去当前柱子的高度,作为当前宽度装水的实际容量。需要注意的是如果遍历到的柱子,其两侧木板都没有自身高,当前宽度装水的容量是0,而不是一个负数。
class Solution {
public int trap(int[] height) {
int sum = 0;
for (int i = 1; i < height.length - 1; i++) {
int rHeight = height[i];
int lHeight = height[i];
for (int r = i + 1; r < height.length; r++) {
if (height[r] > rHeight) rHeight = height[r];
}
for (int l = i - 1; l >= 0; l--) {
if (height[l] > lHeight) lHeight = height[l];
}
int h = Math.min(lHeight,rHeight) - height[i];
if (h > 0) sum += h;
}
return sum;
}
}
然后我们发现,在遍历元素的过程中,我们每遍历一个柱子,都往两侧找最高的边,但在遍历其中一个“木桶”时,两侧的最大高度显然是不变的,但我们还是每次都要重找一遍,显然做了很多无用功。虽然可以存储木桶的两侧的值再加一堆判断来简化,但既然都做到这一步了,不如直接动态规划:
class Solution {
public int trap(int[] height) {
int sum = 0;
int len = height.length;
int[] maxLeft = new int[len];
int[] maxRight = new int[len];
maxLeft[0] = height[0];
for(int i = 1; i < len; i++) {
maxLeft[i] = Math.max(height[i], maxLeft[i - 1]);
}
maxRight[len - 1] = height[len - 1];
for(int i = len - 2; i >= 0; i--) {
maxRight[i] = Math.max(height[i], maxRight[i + 1]);
}
for (int i = 0; i < len; i++) {
int count = Math.min(maxLeft[i], maxRight[i]) - height[i];
if (count > 0) sum += count;
}
return sum;
}
}
其中maxLeft和maxRight记录的其实就是当前元素所在的“木桶”的两边的高度。
然后就是单调栈方法。说实话很不好想,而且思路和双指针、动态规划完全不一样。双指针、动态规划的思路是找出“木桶”的两侧再计算“桶”的容积,而单调栈方法类似于“填坑”,也就是出现一个凹陷就计算这个凹陷的容积,再把这个凹陷填上,直至填满所有凹陷,再把这些凹陷的容积相加。单调栈栈底到栈顶应该是由大到小。凹陷的高度取决于三个值,待操作元素(出现比栈顶元素更大的值了!)、栈顶元素、栈顶元素下的元素(把栈底到栈顶视为从左到右,该元素是栈顶的左边的元素),坑的高度为待操作元素和栈顶元素下的元素较小的那个,减去栈顶元素(不算已被填平的坑的高度),坑的宽度为待操作元素-栈顶元素下的元素-1。当然填坑只是比较形象的说法,实际实现并没有什么填坑的操作,只是把已经计算完对应水高度的宽度在计算时忽略掉(已经计算完容积并从栈中弹出)。
class Solution {
public int trap(int[] height) {
int sum = 0;
int len = height.length;
Deque<Integer> stack = new LinkedList<>();
stack.push(0);
for (int i = 1; i < len; i++) {
if (height[i] < height[stack.peek()]) {
stack.push(i);
}
else if (height[i] == height[stack.peek()]) {
stack.pop();
stack.push(i);
}
else {
while (!stack.isEmpty() && height[i] > height[stack.peek()]) {
int mid = stack.pop();
if (!stack.isEmpty()) {
int left = stack.peek();
int h = Math.min(height[left], height[i]) - height[mid];
int w = i - left - 1;
sum += h * w;
}
}
stack.push(i);
}
}
return sum;
}
}
精简一下:
class Solution {
public int trap(int[] height) {
int sum = 0;
int len = height.length;
Deque<Integer> stack = new LinkedList<>();
stack.push(0);
for (int i = 1; i < len; i++) {
while (!stack.isEmpty() && height[i] > height[stack.peek()]) {
int mid = stack.pop();
if (!stack.isEmpty()) {
int left = stack.peek();
int h = Math.min(height[left], height[i]) - height[mid];
int w = i - left - 1;
sum += h * w;
}
}
stack.push(i);
}
return sum;
}
}