503.下一个更大元素II,42. 接雨水
创始人
2024-04-29 13:59:31
0

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 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 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 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;}
}

相关内容

热门资讯

监控摄像头接入GB28181平... 流程简介将监控摄像头的视频在网站和APP中直播,要解决的几个问题是:1&...
Windows10添加群晖磁盘... 在使用群晖NAS时,我们需要通过本地映射的方式把NAS映射成本地的一块磁盘使用。 通过...
protocol buffer... 目录 目录 什么是protocol buffer 1.protobuf 1.1安装  1.2使用...
在Word、WPS中插入AxM... 引言 我最近需要写一些文章,在排版时发现AxMath插入的公式竟然会导致行间距异常&#...
【PdgCntEditor】解... 一、问题背景 大部分的图书对应的PDF,目录中的页码并非PDF中直接索引的页码...
Fluent中创建监测点 1 概述某些仿真问题,需要创建监测点,用于获取空间定点的数据࿰...
educoder数据结构与算法...                                                   ...
MySQL下载和安装(Wind... 前言:刚换了一台电脑,里面所有东西都需要重新配置,习惯了所...
修复 爱普生 EPSON L4... L4151 L4153 L4156 L4158 L4163 L4165 L4166 L4168 L4...
MFC文件操作  MFC提供了一个文件操作的基类CFile,这个类提供了一个没有缓存的二进制格式的磁盘...